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

Class LinkPredictionTrain

java.lang.Object

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

public final class LinkPredictionTrain
extends java.lang.Object

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method	Description
LinkPredictionTrainResult	compute()
static org.neo4j.gds.core.utils.mem.MemoryEstimation	estimate(LinkPredictionTrainingPipeline pipeline, LinkPredictionTrainConfig trainConfig)
static java.util.List<org.neo4j.gds.core.utils.progress.tasks.Task>	progressTasks(long relationshipCount, LinkPredictionSplitConfig splitConfig, int numberOfModelSelectionTrials)

Methods inherited from class java.lang.Object

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

Constructor Detail

LinkPredictionTrain

public LinkPredictionTrain(org.neo4j.gds.api.Graph trainGraph,
                           org.neo4j.gds.api.Graph validationGraph,
                           LinkPredictionTrainingPipeline pipeline,
                           LinkPredictionTrainConfig config,
                           org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker,
                           org.neo4j.gds.core.utils.TerminationFlag terminationFlag)

Method Detail

progressTasks

public static java.util.List<org.neo4j.gds.core.utils.progress.tasks.Task> progressTasks(long relationshipCount,
                                                                                         LinkPredictionSplitConfig splitConfig,
                                                                                         int numberOfModelSelectionTrials)

compute

public LinkPredictionTrainResult compute()

estimate

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