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.logisticregression.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

    • 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.ml.nodemodels.logisticregression.NodeLogisticRegressionData,​org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineTrainConfig,​org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineModelInfo> model, NodeClassificationPredictPipelineBaseConfig configuration)  
      protected org.neo4j.gds.ml.nodemodels.logisticregression.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, 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

    • Method Detail

      • estimate

        public static org.neo4j.gds.core.utils.mem.MemoryEstimation estimate​(org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.nodemodels.logisticregression.NodeLogisticRegressionData,​org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineTrainConfig,​org.neo4j.gds.ml.pipeline.nodePipeline.NodeClassificationPipelineModelInfo> model,
                                                                     NodeClassificationPredictPipelineBaseConfig configuration)

      • 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.logisticregression.NodeClassificationResult>

      • execute

        protected org.neo4j.gds.ml.nodemodels.logisticregression.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.logisticregression.NodeClassificationResult>