Package org.neo4j.gds.ml.nodemodels

Class NodeClassificationTrainPipelineAlgorithmFactory

java.lang.Object

org.neo4j.gds.GraphStoreAlgorithmFactory<NodeClassificationTrainPipelineExecutor,org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig>

org.neo4j.gds.ml.nodemodels.NodeClassificationTrainPipelineAlgorithmFactory

All Implemented Interfaces:

org.neo4j.gds.AlgorithmFactory<org.neo4j.gds.api.GraphStore,NodeClassificationTrainPipelineExecutor,org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig>

public class NodeClassificationTrainPipelineAlgorithmFactory
extends org.neo4j.gds.GraphStoreAlgorithmFactory<NodeClassificationTrainPipelineExecutor,org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig>

Nested Class Summary

Nested classes/interfaces inherited from interface org.neo4j.gds.AlgorithmFactory

org.neo4j.gds.AlgorithmFactory.Visitor<ALGO extends org.neo4j.gds.Algorithm<?>,CONFIG extends org.neo4j.gds.config.AlgoBaseConfig>

Constructor Summary

Constructors

Constructor	Description
NodeClassificationTrainPipelineAlgorithmFactory(org.neo4j.gds.executor.ExecutionContext executionContext, org.neo4j.gds.core.model.ModelCatalog modelCatalog)

Method Summary

Modifier and Type	Method	Description
NodeClassificationTrainPipelineExecutor	build(org.neo4j.gds.api.GraphStore graphStore, org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig configuration, org.neo4j.gds.core.utils.mem.AllocationTracker allocationTracker, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)
org.neo4j.gds.core.utils.mem.MemoryEstimation	memoryEstimation(org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig configuration)
org.neo4j.gds.core.utils.progress.tasks.Task	progressTask(org.neo4j.gds.api.GraphStore graphStore, org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig config)
java.lang.String	taskName()

Methods inherited from class org.neo4j.gds.GraphStoreAlgorithmFactory

accept

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

build, build

Constructor Detail

NodeClassificationTrainPipelineAlgorithmFactory

public NodeClassificationTrainPipelineAlgorithmFactory(org.neo4j.gds.executor.ExecutionContext executionContext,
                                                       org.neo4j.gds.core.model.ModelCatalog modelCatalog)

Method Detail

build

public NodeClassificationTrainPipelineExecutor build(org.neo4j.gds.api.GraphStore graphStore,
                                                     org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig configuration,
                                                     org.neo4j.gds.core.utils.mem.AllocationTracker allocationTracker,
                                                     org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

memoryEstimation

public org.neo4j.gds.core.utils.mem.MemoryEstimation memoryEstimation(org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig configuration)

taskName

public java.lang.String taskName()

progressTask

public org.neo4j.gds.core.utils.progress.tasks.Task progressTask(org.neo4j.gds.api.GraphStore graphStore,
                                                                 org.neo4j.gds.ml.nodemodels.pipeline.NodeClassificationPipelineTrainConfig config)