Package org.neo4j.gds.ml.pipeline.node.classification.predict

Class NodeClassificationPredictPipelineExecutor

java.lang.Object

org.neo4j.gds.Algorithm<RESULT>

org.neo4j.gds.ml.pipeline.PredictPipelineExecutor<NodeClassificationPredictPipelineBaseConfig,org.neo4j.gds.ml.pipeline.nodePipeline.NodePropertyPredictPipeline,NodeClassificationPredictPipelineExecutor.NodeClassificationPipelineResult>

org.neo4j.gds.ml.pipeline.node.classification.predict.NodeClassificationPredictPipelineExecutor

public class NodeClassificationPredictPipelineExecutor
extends org.neo4j.gds.ml.pipeline.PredictPipelineExecutor<NodeClassificationPredictPipelineBaseConfig,org.neo4j.gds.ml.pipeline.nodePipeline.NodePropertyPredictPipeline,NodeClassificationPredictPipelineExecutor.NodeClassificationPipelineResult>

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static interface	NodeClassificationPredictPipelineExecutor.NodeClassificationPipelineResult

Field Summary

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

config, executionContext, graphStore, pipeline

Fields inherited from class org.neo4j.gds.Algorithm

progressTracker, terminationFlag

Constructor Summary

Constructors

Constructor

Description

NodeClassificationPredictPipelineExecutor(org.neo4j.gds.ml.pipeline.nodePipeline.NodePropertyPredictPipeline pipeline, NodeClassificationPredictPipelineBaseConfig config, org.neo4j.gds.executor.ExecutionContext executionContext, org.neo4j.gds.api.GraphStore graphStore, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker, org.neo4j.gds.ml.models.Classifier.ClassifierData modelData, org.neo4j.gds.ml.core.subgraph.LocalIdMap classIdMap)

Method Summary

Modifier and Type	Method	Description
static org.neo4j.gds.core.utils.mem.MemoryEstimation	estimate(org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.models.Classifier.ClassifierData,org.neo4j.gds.ml.pipeline.nodePipeline.classification.train.NodeClassificationPipelineTrainConfig,org.neo4j.gds.ml.pipeline.nodePipeline.classification.train.NodeClassificationPipelineModelInfo> model, NodeClassificationPredictPipelineBaseConfig configuration, org.neo4j.gds.core.model.ModelCatalog modelCatalog)
protected NodeClassificationPredictPipelineExecutor.NodeClassificationPipelineResult	execute()
protected org.neo4j.gds.ml.pipeline.PipelineGraphFilter	nodePropertyStepFilter()
static org.neo4j.gds.core.utils.progress.tasks.Task	progressTask(java.lang.String taskName, org.neo4j.gds.ml.pipeline.nodePipeline.NodePropertyPredictPipeline pipeline, org.neo4j.gds.api.GraphStore graphStore)

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

compute, release

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

Constructor Detail

NodeClassificationPredictPipelineExecutor

public NodeClassificationPredictPipelineExecutor(org.neo4j.gds.ml.pipeline.nodePipeline.NodePropertyPredictPipeline pipeline,
                                                 NodeClassificationPredictPipelineBaseConfig config,
                                                 org.neo4j.gds.executor.ExecutionContext executionContext,
                                                 org.neo4j.gds.api.GraphStore graphStore,
                                                 org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker,
                                                 org.neo4j.gds.ml.models.Classifier.ClassifierData modelData,
                                                 org.neo4j.gds.ml.core.subgraph.LocalIdMap classIdMap)

Method Detail

progressTask

public static org.neo4j.gds.core.utils.progress.tasks.Task progressTask(java.lang.String taskName,
                                                                        org.neo4j.gds.ml.pipeline.nodePipeline.NodePropertyPredictPipeline pipeline,
                                                                        org.neo4j.gds.api.GraphStore graphStore)

estimate

public static org.neo4j.gds.core.utils.mem.MemoryEstimation estimate(org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.models.Classifier.ClassifierData,org.neo4j.gds.ml.pipeline.nodePipeline.classification.train.NodeClassificationPipelineTrainConfig,org.neo4j.gds.ml.pipeline.nodePipeline.classification.train.NodeClassificationPipelineModelInfo> model,
                                                                     NodeClassificationPredictPipelineBaseConfig configuration,
                                                                     org.neo4j.gds.core.model.ModelCatalog modelCatalog)

nodePropertyStepFilter

protected org.neo4j.gds.ml.pipeline.PipelineGraphFilter nodePropertyStepFilter()

Specified by:

nodePropertyStepFilter in class org.neo4j.gds.ml.pipeline.PredictPipelineExecutor<NodeClassificationPredictPipelineBaseConfig,org.neo4j.gds.ml.pipeline.nodePipeline.NodePropertyPredictPipeline,NodeClassificationPredictPipelineExecutor.NodeClassificationPipelineResult>

execute

protected NodeClassificationPredictPipelineExecutor.NodeClassificationPipelineResult execute()

Specified by:

execute in class org.neo4j.gds.ml.pipeline.PredictPipelineExecutor<NodeClassificationPredictPipelineBaseConfig,org.neo4j.gds.ml.pipeline.nodePipeline.NodePropertyPredictPipeline,NodeClassificationPredictPipelineExecutor.NodeClassificationPipelineResult>