Package org.neo4j.gds.ml.pipeline

Class PipelineExecutor<PIPELINE_CONFIG extends org.neo4j.gds.config.AlgoBaseConfig & org.neo4j.gds.config.GraphNameConfig,PIPELINE extends Pipeline<?>,RESULT>

java.lang.Object

org.neo4j.gds.Algorithm<RESULT>

org.neo4j.gds.ml.pipeline.PipelineExecutor<PIPELINE_CONFIG,PIPELINE,RESULT>

Direct Known Subclasses:

LinkPredictionTrainPipelineExecutor

public abstract class PipelineExecutor<PIPELINE_CONFIG extends org.neo4j.gds.config.AlgoBaseConfig & org.neo4j.gds.config.GraphNameConfig,PIPELINE extends Pipeline<?>,RESULT>
extends org.neo4j.gds.Algorithm<RESULT>

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	PipelineExecutor.DatasetSplits

Field Summary

Fields

Modifier and Type	Field	Description
protected PIPELINE_CONFIG	config
protected org.neo4j.gds.executor.ExecutionContext	executionContext
protected org.neo4j.gds.api.GraphStore	graphStore
protected PIPELINE	pipeline
protected org.neo4j.gds.api.schema.GraphSchema	schemaBeforeSteps

Fields inherited from class org.neo4j.gds.Algorithm

progressTracker, terminationFlag

Constructor Summary

Constructors

Modifier	Constructor	Description
protected	PipelineExecutor(PIPELINE pipeline, PIPELINE_CONFIG config, org.neo4j.gds.executor.ExecutionContext executionContext, org.neo4j.gds.api.GraphStore graphStore, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

Method Summary

Modifier and Type	Method	Description
protected void	additionalGraphStoreCleanup(java.util.Map<PipelineExecutor.DatasetSplits,PipelineGraphFilter> datasets)
RESULT	compute()
protected abstract RESULT	execute(java.util.Map<PipelineExecutor.DatasetSplits,PipelineGraphFilter> dataSplits)
abstract java.util.Map<PipelineExecutor.DatasetSplits,PipelineGraphFilter>	generateDatasetSplitGraphFilters()
protected abstract java.util.Set<org.neo4j.gds.RelationshipType>	getAvailableRelTypesForNodePropertySteps()
abstract void	splitDatasets()

Methods inherited from class org.neo4j.gds.Algorithm

getProgressTracker, getTerminationFlag, setTerminationFlag

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

pipeline

protected final PIPELINE extends Pipeline<?> pipeline

config

protected final PIPELINE_CONFIG extends org.neo4j.gds.config.AlgoBaseConfig & org.neo4j.gds.config.GraphNameConfig config

executionContext

protected final org.neo4j.gds.executor.ExecutionContext executionContext

graphStore

protected final org.neo4j.gds.api.GraphStore graphStore

schemaBeforeSteps

protected final org.neo4j.gds.api.schema.GraphSchema schemaBeforeSteps

Constructor Detail

PipelineExecutor

protected PipelineExecutor(PIPELINE pipeline,
                           PIPELINE_CONFIG config,
                           org.neo4j.gds.executor.ExecutionContext executionContext,
                           org.neo4j.gds.api.GraphStore graphStore,
                           org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

Method Detail

generateDatasetSplitGraphFilters

public abstract java.util.Map<PipelineExecutor.DatasetSplits,PipelineGraphFilter> generateDatasetSplitGraphFilters()

splitDatasets

public abstract void splitDatasets()

execute

protected abstract RESULT execute(java.util.Map<PipelineExecutor.DatasetSplits,PipelineGraphFilter> dataSplits)

compute

public RESULT compute()

Specified by:

compute in class org.neo4j.gds.Algorithm<RESULT>

getAvailableRelTypesForNodePropertySteps

protected abstract java.util.Set<org.neo4j.gds.RelationshipType> getAvailableRelTypesForNodePropertySteps()

additionalGraphStoreCleanup

protected void additionalGraphStoreCleanup(java.util.Map<PipelineExecutor.DatasetSplits,PipelineGraphFilter> datasets)