Package org.neo4j.gds.ml.nodemodels.pipeline

Class NodeClassificationTrainPipelineExecutor

java.lang.Object

org.neo4j.gds.Algorithm<RESULT>

org.neo4j.gds.ml.pipeline.PipelineExecutor<org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig,NodeClassificationPipeline,NodeClassificationPipelineTrainResult>

org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationTrainPipelineExecutor

All Implemented Interfaces:

org.neo4j.gds.core.utils.TerminationFlag

public class NodeClassificationTrainPipelineExecutor
extends org.neo4j.gds.ml.pipeline.PipelineExecutor<org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig,NodeClassificationPipeline,NodeClassificationPipelineTrainResult>

Nested Class Summary

Nested classes/interfaces inherited from class org.neo4j.gds.ml.pipeline.PipelineExecutor

org.neo4j.gds.ml.pipeline.PipelineExecutor.DatasetSplits, org.neo4j.gds.ml.pipeline.PipelineExecutor.GraphFilter

Field Summary

Fields

Modifier and Type	Field	Description
static java.lang.String	MODEL_TYPE

Fields inherited from class org.neo4j.gds.ml.pipeline.PipelineExecutor

config, executionContext, graphName, graphStore, pipeline

Fields inherited from class org.neo4j.gds.Algorithm

progressTracker, terminationFlag

Fields inherited from interface org.neo4j.gds.core.utils.TerminationFlag

RUN_CHECK_NODE_COUNT, RUNNING_TRUE

Constructor Summary

Constructors

Constructor	Description
NodeClassificationTrainPipelineExecutor(NodeClassificationPipeline pipeline, org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig config, org.neo4j.gds.executor.ExecutionContext executionContext, org.neo4j.gds.api.GraphStore graphStore, java.lang.String graphName, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

Method Summary

Modifier and Type	Method	Description
protected NodeClassificationPipelineTrainResult	execute(java.util.Map<org.neo4j.gds.ml.pipeline.PipelineExecutor.DatasetSplits,org.neo4j.gds.ml.pipeline.PipelineExecutor.GraphFilter> dataSplits)
java.util.Map<org.neo4j.gds.ml.pipeline.PipelineExecutor.DatasetSplits,org.neo4j.gds.ml.pipeline.PipelineExecutor.GraphFilter>	splitDataset()

Methods inherited from class org.neo4j.gds.ml.pipeline.PipelineExecutor

cleanUpGraphStore, compute, release, validate

Methods inherited from class org.neo4j.gds.Algorithm

getProgressTracker, getTerminationFlag, running, setTerminationFlag

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.neo4j.gds.core.utils.TerminationFlag

assertRunning

Field Detail

MODEL_TYPE

public static final java.lang.String MODEL_TYPE

See Also:

Constant Field Values

Constructor Detail

NodeClassificationTrainPipelineExecutor

public NodeClassificationTrainPipelineExecutor(NodeClassificationPipeline pipeline,
                                               org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig config,
                                               org.neo4j.gds.executor.ExecutionContext executionContext,
                                               org.neo4j.gds.api.GraphStore graphStore,
                                               java.lang.String graphName,
                                               org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

Method Detail

splitDataset

public java.util.Map<org.neo4j.gds.ml.pipeline.PipelineExecutor.DatasetSplits,org.neo4j.gds.ml.pipeline.PipelineExecutor.GraphFilter> splitDataset()

Specified by:

splitDataset in class org.neo4j.gds.ml.pipeline.PipelineExecutor<org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig,NodeClassificationPipeline,NodeClassificationPipelineTrainResult>

execute

protected NodeClassificationPipelineTrainResult execute(java.util.Map<org.neo4j.gds.ml.pipeline.PipelineExecutor.DatasetSplits,org.neo4j.gds.ml.pipeline.PipelineExecutor.GraphFilter> dataSplits)

Specified by:

execute in class org.neo4j.gds.ml.pipeline.PipelineExecutor<org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig,NodeClassificationPipeline,NodeClassificationPipelineTrainResult>