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

Class NodeClassificationPredictPipelineExecutor

  • All Implemented Interfaces:
    org.neo4j.gds.core.utils.TerminationFlag
    public class NodeClassificationPredictPipelineExecutor
extends org.neo4j.gds.ml.pipeline.PipelineExecutor<NodeClassificationPredictPipelineBaseConfig,​org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipeline,​org.neo4j.gds.ml.nodemodels.NodeClassificationPredict.NodeClassificationResult>

    • 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
      NodeClassificationPredictPipelineExecutor​(org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipeline pipeline, NodeClassificationPredictPipelineBaseConfig 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, org.neo4j.gds.models.Classifier.ClassifierData modelData)  

    • 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.core.model.Model<org.neo4j.gds.models.Classifier.ClassifierData,​org.neo4j.gds.ml.pipeline.nodePipeline.train.NodeClassificationPipelineTrainConfig,​org.neo4j.gds.ml.pipeline.nodePipeline.train.NodeClassificationPipelineModelInfo> model, NodeClassificationPredictPipelineBaseConfig configuration, org.neo4j.gds.core.model.ModelCatalog modelCatalog)  
      protected org.neo4j.gds.ml.nodemodels.NodeClassificationPredict.NodeClassificationResult 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

      • NodeClassificationPredictPipelineExecutor

        public NodeClassificationPredictPipelineExecutor​(org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipeline pipeline,
                                                 NodeClassificationPredictPipelineBaseConfig 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,
                                                 org.neo4j.gds.models.Classifier.ClassifierData modelData)

    • Method Detail

      • estimate

        public static org.neo4j.gds.core.utils.mem.MemoryEstimation estimate​(org.neo4j.gds.core.model.Model<org.neo4j.gds.models.Classifier.ClassifierData,​org.neo4j.gds.ml.pipeline.nodePipeline.train.NodeClassificationPipelineTrainConfig,​org.neo4j.gds.ml.pipeline.nodePipeline.train.NodeClassificationPipelineModelInfo> model,
                                                                     NodeClassificationPredictPipelineBaseConfig 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<NodeClassificationPredictPipelineBaseConfig,​org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipeline,​org.neo4j.gds.ml.nodemodels.NodeClassificationPredict.NodeClassificationResult>

      • execute

        protected org.neo4j.gds.ml.nodemodels.NodeClassificationPredict.NodeClassificationResult 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<NodeClassificationPredictPipelineBaseConfig,​org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipeline,​org.neo4j.gds.ml.nodemodels.NodeClassificationPredict.NodeClassificationResult>