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

Class NodeClassificationTrain

java.lang.Object

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

All Implemented Interfaces:

PipelineTrainer<NodeClassificationTrainResult>

public final class NodeClassificationTrain
extends java.lang.Object
implements PipelineTrainer<NodeClassificationTrainResult>

Method Summary

Modifier and Type	Method	Description
static NodeClassificationTrain	create(org.neo4j.gds.api.GraphStore graphStore, NodeClassificationTrainingPipeline pipeline, NodeClassificationPipelineTrainConfig config, NodeFeatureProducer<NodeClassificationPipelineTrainConfig> nodeFeatureProducer, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)
static org.neo4j.gds.core.utils.mem.MemoryEstimation	estimate(NodeClassificationTrainingPipeline pipeline, NodeClassificationPipelineTrainConfig configuration, org.neo4j.gds.core.model.ModelCatalog modelCatalog)
static org.neo4j.gds.core.utils.progress.tasks.Task	progressTask(NodeClassificationTrainingPipeline pipeline, long nodeCount)
NodeClassificationTrainResult	run()
void	setTerminationFlag(org.neo4j.gds.core.utils.TerminationFlag terminationFlag)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Method Detail

estimate

public static org.neo4j.gds.core.utils.mem.MemoryEstimation estimate(NodeClassificationTrainingPipeline pipeline,
                                                                     NodeClassificationPipelineTrainConfig configuration,
                                                                     org.neo4j.gds.core.model.ModelCatalog modelCatalog)

progressTask

public static org.neo4j.gds.core.utils.progress.tasks.Task progressTask(NodeClassificationTrainingPipeline pipeline,
                                                                        long nodeCount)

create

public static NodeClassificationTrain create(org.neo4j.gds.api.GraphStore graphStore,
                                             NodeClassificationTrainingPipeline pipeline,
                                             NodeClassificationPipelineTrainConfig config,
                                             NodeFeatureProducer<NodeClassificationPipelineTrainConfig> nodeFeatureProducer,
                                             org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

setTerminationFlag

public void setTerminationFlag(org.neo4j.gds.core.utils.TerminationFlag terminationFlag)

Specified by:

setTerminationFlag in interface PipelineTrainer<NodeClassificationTrainResult>

run

public NodeClassificationTrainResult run()

Specified by:

run in interface PipelineTrainer<NodeClassificationTrainResult>