Package org.neo4j.gds.ml.linkmodels.pipeline.train

Class LinkPredictionTrain

java.lang.Object

org.neo4j.gds.Algorithm<LinkPredictionTrainResult>

org.neo4j.gds.ml.linkmodels.pipeline.train.LinkPredictionTrain

All Implemented Interfaces:

org.neo4j.gds.core.utils.TerminationFlag

public class LinkPredictionTrain
extends org.neo4j.gds.Algorithm<LinkPredictionTrainResult>

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static interface	LinkPredictionTrain.ModelSelectResult

Field Summary

Fields

Modifier and Type	Field	Description
static java.lang.String	MODEL_TYPE

Fields inherited from class org.neo4j.gds.Algorithm

progressTracker, terminationFlag

Fields inherited from interface org.neo4j.gds.core.utils.TerminationFlag

RUN_CHECK_NODE_COUNT, RUNNING_TRUE

Constructor Summary

Constructors

Constructor	Description
LinkPredictionTrain(org.neo4j.gds.api.Graph trainGraph, org.neo4j.gds.api.Graph validationGraph, org.neo4j.gds.ml.pipeline.linkPipeline.LinkPredictionPipeline pipeline, LinkPredictionTrainConfig trainConfig, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

Method Summary

Modifier and Type	Method	Description
LinkPredictionTrainResult	compute()
void	release()

Methods inherited from class org.neo4j.gds.Algorithm

getProgressTracker, getTerminationFlag, running, setTerminationFlag

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.core.utils.TerminationFlag

assertRunning

Field Detail

MODEL_TYPE

public static final java.lang.String MODEL_TYPE

See Also:

Constant Field Values

Constructor Detail

LinkPredictionTrain

public LinkPredictionTrain(org.neo4j.gds.api.Graph trainGraph,
                           org.neo4j.gds.api.Graph validationGraph,
                           org.neo4j.gds.ml.pipeline.linkPipeline.LinkPredictionPipeline pipeline,
                           LinkPredictionTrainConfig trainConfig,
                           org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

Method Detail

compute

public LinkPredictionTrainResult compute()

Specified by:

compute in class org.neo4j.gds.Algorithm<LinkPredictionTrainResult>

release

public void release()

Specified by:

release in class org.neo4j.gds.Algorithm<LinkPredictionTrainResult>