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

Class LinkPredictionTrainFactory

java.lang.Object

org.neo4j.gds.AlgorithmFactory<LinkPredictionTrain,LinkPredictionTrainConfig>

org.neo4j.gds.ml.linkmodels.pipeline.LinkPredictionTrainFactory

public class LinkPredictionTrainFactory
extends org.neo4j.gds.AlgorithmFactory<LinkPredictionTrain,LinkPredictionTrainConfig>

Constructor Summary

Constructors

Constructor	Description
LinkPredictionTrainFactory(org.neo4j.kernel.database.NamedDatabaseId databaseId, org.neo4j.gds.BaseProc caller)

Method Summary

Modifier and Type	Method	Description
LinkPredictionTrain	build(org.neo4j.gds.api.Graph graph, LinkPredictionTrainConfig trainConfig, org.neo4j.gds.core.utils.mem.AllocationTracker allocationTracker, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)
org.neo4j.gds.core.utils.mem.MemoryEstimation	memoryEstimation(LinkPredictionTrainConfig configuration)
org.neo4j.gds.core.utils.progress.tasks.Task	progressTask(org.neo4j.gds.api.Graph graph, LinkPredictionTrainConfig config)
protected java.lang.String	taskName()

Methods inherited from class org.neo4j.gds.AlgorithmFactory

build

Methods inherited from class java.lang.Object

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

Constructor Detail

LinkPredictionTrainFactory

public LinkPredictionTrainFactory(org.neo4j.kernel.database.NamedDatabaseId databaseId,
                                  org.neo4j.gds.BaseProc caller)

Method Detail

build

public LinkPredictionTrain build(org.neo4j.gds.api.Graph graph,
                                 LinkPredictionTrainConfig trainConfig,
                                 org.neo4j.gds.core.utils.mem.AllocationTracker allocationTracker,
                                 org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

Specified by:

build in class org.neo4j.gds.AlgorithmFactory<LinkPredictionTrain,LinkPredictionTrainConfig>

taskName

protected java.lang.String taskName()

Specified by:

taskName in class org.neo4j.gds.AlgorithmFactory<LinkPredictionTrain,LinkPredictionTrainConfig>

progressTask

public org.neo4j.gds.core.utils.progress.tasks.Task progressTask(org.neo4j.gds.api.Graph graph,
                                                                 LinkPredictionTrainConfig config)

Overrides:

progressTask in class org.neo4j.gds.AlgorithmFactory<LinkPredictionTrain,LinkPredictionTrainConfig>

memoryEstimation

public org.neo4j.gds.core.utils.mem.MemoryEstimation memoryEstimation(LinkPredictionTrainConfig configuration)

Overrides:

memoryEstimation in class org.neo4j.gds.AlgorithmFactory<LinkPredictionTrain,LinkPredictionTrainConfig>