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

Class LinkPredictionTrainPipelineAlgorithmFactory

java.lang.Object

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

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

All Implemented Interfaces:

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

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

Nested Class Summary

Nested classes/interfaces inherited from interface org.neo4j.gds.AlgorithmFactory

org.neo4j.gds.AlgorithmFactory.Visitor<ALGO extends org.neo4j.gds.Algorithm<?>,CONFIG extends org.neo4j.gds.config.AlgoBaseConfig>

Constructor Summary

Constructors

Constructor	Description
LinkPredictionTrainPipelineAlgorithmFactory(org.neo4j.gds.executor.ExecutionContext executionContext, org.neo4j.gds.core.model.ModelCatalog modelCatalog)

Method Summary

Modifier and Type	Method	Description
LinkPredictionTrainPipelineExecutor	build(org.neo4j.gds.api.GraphStore graphStore, 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.GraphStore graphStore, LinkPredictionTrainConfig config)
java.lang.String	taskName()

Methods inherited from class org.neo4j.gds.GraphStoreAlgorithmFactory

accept

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.AlgorithmFactory

build, build

Constructor Detail

LinkPredictionTrainPipelineAlgorithmFactory

public LinkPredictionTrainPipelineAlgorithmFactory(org.neo4j.gds.executor.ExecutionContext executionContext,
                                                   org.neo4j.gds.core.model.ModelCatalog modelCatalog)

Method Detail

build

public LinkPredictionTrainPipelineExecutor build(org.neo4j.gds.api.GraphStore graphStore,
                                                 LinkPredictionTrainConfig trainConfig,
                                                 org.neo4j.gds.core.utils.mem.AllocationTracker allocationTracker,
                                                 org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

taskName

public java.lang.String taskName()

progressTask

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

memoryEstimation

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