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

Class LinkPredictionTrain

java.lang.Object

org.neo4j.gds.Algorithm<LinkPredictionTrainResult>

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

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

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method	Description
LinkPredictionTrainResult	compute()
static org.neo4j.gds.core.utils.mem.MemoryEstimation	estimate(LinkPredictionPipeline pipeline, LinkPredictionTrainConfig trainConfig)
static org.neo4j.gds.ml.core.subgraph.LocalIdMap	makeClassIdMap()
static org.neo4j.gds.core.utils.progress.tasks.Task	progressTask()
void	release()

Methods inherited from class org.neo4j.gds.Algorithm

getProgressTracker, getTerminationFlag, setTerminationFlag

Methods inherited from class java.lang.Object

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

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,
                           LinkPredictionPipeline pipeline,
                           LinkPredictionTrainConfig config,
                           org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

Method Detail

makeClassIdMap

public static org.neo4j.gds.ml.core.subgraph.LocalIdMap makeClassIdMap()

progressTask

public static org.neo4j.gds.core.utils.progress.tasks.Task progressTask()

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>

estimate

public static org.neo4j.gds.core.utils.mem.MemoryEstimation estimate(LinkPredictionPipeline pipeline,
                                                                     LinkPredictionTrainConfig trainConfig)