Package org.neo4j.gds.ml.pipeline.nodePipeline.classification.train

Class NodeClassificationTrainPipelineAlgorithmFactory

java.lang.Object

org.neo4j.gds.GraphStoreAlgorithmFactory<NodeClassificationTrainAlgorithm,NodeClassificationPipelineTrainConfig>

org.neo4j.gds.ml.pipeline.nodePipeline.classification.train.NodeClassificationTrainPipelineAlgorithmFactory

All Implemented Interfaces:

org.neo4j.gds.AlgorithmFactory<org.neo4j.gds.api.GraphStore,NodeClassificationTrainAlgorithm,NodeClassificationPipelineTrainConfig>

public class NodeClassificationTrainPipelineAlgorithmFactory
extends org.neo4j.gds.GraphStoreAlgorithmFactory<NodeClassificationTrainAlgorithm,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)

Method Summary

Modifier and Type	Method	Description
NodeClassificationTrainAlgorithm	build(org.neo4j.gds.api.GraphStore graphStore, NodeClassificationPipelineTrainConfig configuration, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)
NodeClassificationTrainAlgorithm	build(org.neo4j.gds.api.GraphStore graphStore, NodeClassificationPipelineTrainConfig configuration, NodeClassificationTrainingPipeline pipeline, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)
org.neo4j.gds.core.utils.mem.MemoryEstimation	memoryEstimation(NodeClassificationPipelineTrainConfig configuration)
static org.neo4j.gds.core.utils.progress.tasks.Task	progressTask(org.neo4j.gds.api.GraphStore graphStore, NodeClassificationTrainingPipeline pipeline)
org.neo4j.gds.core.utils.progress.tasks.Task	progressTask(org.neo4j.gds.api.GraphStore graphStore, 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, estimatedGraphDimensionTransformer

Constructor Detail

NodeClassificationTrainPipelineAlgorithmFactory

public NodeClassificationTrainPipelineAlgorithmFactory(org.neo4j.gds.executor.ExecutionContext executionContext)

Method Detail

build

public NodeClassificationTrainAlgorithm build(org.neo4j.gds.api.GraphStore graphStore,
                                              NodeClassificationPipelineTrainConfig configuration,
                                              org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

build

public NodeClassificationTrainAlgorithm build(org.neo4j.gds.api.GraphStore graphStore,
                                              NodeClassificationPipelineTrainConfig configuration,
                                              NodeClassificationTrainingPipeline pipeline,
                                              org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

memoryEstimation

public org.neo4j.gds.core.utils.mem.MemoryEstimation memoryEstimation(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,
                                                                 NodeClassificationPipelineTrainConfig config)

progressTask

public static org.neo4j.gds.core.utils.progress.tasks.Task progressTask(org.neo4j.gds.api.GraphStore graphStore,
                                                                        NodeClassificationTrainingPipeline pipeline)