Package org.neo4j.gds.ml.pipeline.nodePipeline.regression

Class NodeRegressionTrain

java.lang.Object

org.neo4j.gds.ml.pipeline.nodePipeline.regression.NodeRegressionTrain

All Implemented Interfaces:

PipelineTrainer<NodeRegressionTrainResult>

public final class NodeRegressionTrain
extends java.lang.Object
implements PipelineTrainer<NodeRegressionTrainResult>

Method Summary

Modifier and Type	Method	Description
static NodeRegressionTrain	create(org.neo4j.gds.api.GraphStore graphStore, NodeRegressionTrainingPipeline pipeline, NodeRegressionPipelineTrainConfig config, NodeFeatureProducer<NodeRegressionPipelineTrainConfig> nodeFeatureProducer, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)
static org.neo4j.gds.core.utils.progress.tasks.Task	progressTask(NodePropertyTrainingPipeline pipeline, long nodeCount)
NodeRegressionTrainResult	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

progressTask

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

create

public static NodeRegressionTrain create(org.neo4j.gds.api.GraphStore graphStore,
                                         NodeRegressionTrainingPipeline pipeline,
                                         NodeRegressionPipelineTrainConfig config,
                                         NodeFeatureProducer<NodeRegressionPipelineTrainConfig> 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<NodeRegressionTrainResult>

run

public NodeRegressionTrainResult run()

Specified by:

run in interface PipelineTrainer<NodeRegressionTrainResult>