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

Class LinkPredictionTrainPipelineAlgorithmFactory

java.lang.Object

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

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

public class LinkPredictionTrainPipelineAlgorithmFactory
extends org.neo4j.gds.AlgorithmFactory<LinkPredictionTrainPipelineExecutor,LinkPredictionTrainConfig>

Constructor Summary

Constructors

Constructor	Description
LinkPredictionTrainPipelineAlgorithmFactory(org.neo4j.gds.BaseProc caller, org.neo4j.kernel.database.NamedDatabaseId databaseId, org.neo4j.gds.core.model.ModelCatalog modelCatalog)

Method Summary

Modifier and Type	Method	Description
LinkPredictionTrainPipelineExecutor	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

LinkPredictionTrainPipelineAlgorithmFactory

public LinkPredictionTrainPipelineAlgorithmFactory(org.neo4j.gds.BaseProc caller,
                                                   org.neo4j.kernel.database.NamedDatabaseId databaseId,
                                                   org.neo4j.gds.core.model.ModelCatalog modelCatalog)

Method Detail

build

public LinkPredictionTrainPipelineExecutor 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<LinkPredictionTrainPipelineExecutor,LinkPredictionTrainConfig>

taskName

protected java.lang.String taskName()

Specified by:

taskName in class org.neo4j.gds.AlgorithmFactory<LinkPredictionTrainPipelineExecutor,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<LinkPredictionTrainPipelineExecutor,LinkPredictionTrainConfig>

memoryEstimation

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

Overrides:

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