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

Class NodeClassificationTrainPipelineExecutor

  • All Implemented Interfaces:
    org.neo4j.gds.core.utils.TerminationFlag
    public class NodeClassificationTrainPipelineExecutor
extends org.neo4j.gds.ml.pipeline.PipelineExecutor<org.neo4j.gds.ml.pipeline.nodePipeline.train.NodeClassificationPipelineTrainConfig,​org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipeline,​org.neo4j.gds.ml.pipeline.nodePipeline.train.NodeClassificationTrainResult>

    • 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 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​(org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipeline pipeline, org.neo4j.gds.ml.pipeline.nodePipeline.train.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

      All Methods Static Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      static org.neo4j.gds.core.utils.mem.MemoryEstimation estimate​(org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipeline pipeline, org.neo4j.gds.ml.pipeline.nodePipeline.train.NodeClassificationPipelineTrainConfig configuration, org.neo4j.gds.core.model.ModelCatalog modelCatalog)  
      protected org.neo4j.gds.ml.pipeline.nodePipeline.train.NodeClassificationTrainResult 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, estimateNodePropertySteps, release, validate, validateTrainingParameterSpace

      • 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

    • Constructor Detail

      • NodeClassificationTrainPipelineExecutor

        public NodeClassificationTrainPipelineExecutor​(org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipeline pipeline,
                                               org.neo4j.gds.ml.pipeline.nodePipeline.train.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

      • estimate

        public static org.neo4j.gds.core.utils.mem.MemoryEstimation estimate​(org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipeline pipeline,
                                                                     org.neo4j.gds.ml.pipeline.nodePipeline.train.NodeClassificationPipelineTrainConfig configuration,
                                                                     org.neo4j.gds.core.model.ModelCatalog modelCatalog)

      • 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.pipeline.nodePipeline.train.NodeClassificationPipelineTrainConfig,​org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipeline,​org.neo4j.gds.ml.pipeline.nodePipeline.train.NodeClassificationTrainResult>

      • execute

        protected org.neo4j.gds.ml.pipeline.nodePipeline.train.NodeClassificationTrainResult 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.pipeline.nodePipeline.train.NodeClassificationPipelineTrainConfig,​org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipeline,​org.neo4j.gds.ml.pipeline.nodePipeline.train.NodeClassificationTrainResult>