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

Class LinkPredictionTrainPipelineAlgorithmFactory

java.lang.Object

org.neo4j.gds.GraphStoreAlgorithmFactory<LinkPredictionTrainPipelineExecutor,org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainConfig>

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

All Implemented Interfaces:

org.neo4j.gds.AlgorithmFactory<org.neo4j.gds.api.GraphStore,LinkPredictionTrainPipelineExecutor,org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainConfig>

public class LinkPredictionTrainPipelineAlgorithmFactory
extends org.neo4j.gds.GraphStoreAlgorithmFactory<LinkPredictionTrainPipelineExecutor,org.neo4j.gds.ml.pipeline.linkPipeline.train.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)

Method Summary

Modifier and Type	Method	Description
LinkPredictionTrainPipelineExecutor	build(org.neo4j.gds.api.GraphStore graphStore, org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainConfig trainConfig, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)
org.neo4j.gds.core.GraphDimensions	estimatedGraphDimensionTransformer(org.neo4j.gds.core.GraphDimensions graphDimensions, org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainConfig config)
org.neo4j.gds.core.utils.mem.MemoryEstimation	memoryEstimation(org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainConfig configuration)
org.neo4j.gds.core.utils.progress.tasks.Task	progressTask(org.neo4j.gds.api.GraphStore graphStore, org.neo4j.gds.ml.pipeline.linkPipeline.train.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)

Method Detail

build

public LinkPredictionTrainPipelineExecutor build(org.neo4j.gds.api.GraphStore graphStore,
                                                 org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainConfig trainConfig,
                                                 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,
                                                                 org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainConfig config)

memoryEstimation

public org.neo4j.gds.core.utils.mem.MemoryEstimation memoryEstimation(org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainConfig configuration)

estimatedGraphDimensionTransformer

public org.neo4j.gds.core.GraphDimensions estimatedGraphDimensionTransformer(org.neo4j.gds.core.GraphDimensions graphDimensions,
                                                                             org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainConfig config)