Class LinkPredictionTrainPipelineExecutor
- java.lang.Object
-
- org.neo4j.gds.Algorithm<RESULT>
-
- org.neo4j.gds.ml.pipeline.PipelineExecutor<LinkPredictionTrainConfig,LinkPredictionTrainingPipeline,LinkPredictionTrainPipelineExecutor.LinkPredictionTrainPipelineResult>
-
- org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainPipelineExecutor
-
public class LinkPredictionTrainPipelineExecutor extends PipelineExecutor<LinkPredictionTrainConfig,LinkPredictionTrainingPipeline,LinkPredictionTrainPipelineExecutor.LinkPredictionTrainPipelineResult>
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static interfaceLinkPredictionTrainPipelineExecutor.LinkPredictionTrainPipelineResult-
Nested classes/interfaces inherited from class org.neo4j.gds.ml.pipeline.PipelineExecutor
PipelineExecutor.DatasetSplits
-
-
Field Summary
-
Fields inherited from class org.neo4j.gds.ml.pipeline.PipelineExecutor
config, executionContext, graphStore, pipeline, schemaBeforeSteps
-
-
Constructor Summary
Constructors Constructor Description LinkPredictionTrainPipelineExecutor(LinkPredictionTrainingPipeline pipeline, LinkPredictionTrainConfig config, org.neo4j.gds.executor.ExecutionContext executionContext, org.neo4j.gds.api.GraphStore graphStore, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description protected voidadditionalGraphStoreCleanup(java.util.Map<PipelineExecutor.DatasetSplits,PipelineGraphFilter> datasets)static org.neo4j.gds.core.utils.mem.MemoryEstimationestimate(org.neo4j.gds.executor.ExecutionContext executionContext, LinkPredictionTrainingPipeline pipeline, LinkPredictionTrainConfig configuration)protected LinkPredictionTrainPipelineExecutor.LinkPredictionTrainPipelineResultexecute(java.util.Map<PipelineExecutor.DatasetSplits,PipelineGraphFilter> dataSplits)java.util.Map<PipelineExecutor.DatasetSplits,PipelineGraphFilter>generateDatasetSplitGraphFilters()protected java.util.Set<org.neo4j.gds.RelationshipType>getAvailableRelTypesForNodePropertySteps()static org.neo4j.gds.core.utils.progress.tasks.TaskprogressTask(java.lang.String taskName, LinkPredictionTrainingPipeline pipeline, long relationshipCount)voidsplitDatasets()-
Methods inherited from class org.neo4j.gds.ml.pipeline.PipelineExecutor
compute
-
-
-
-
Constructor Detail
-
LinkPredictionTrainPipelineExecutor
public LinkPredictionTrainPipelineExecutor(LinkPredictionTrainingPipeline pipeline, LinkPredictionTrainConfig config, org.neo4j.gds.executor.ExecutionContext executionContext, org.neo4j.gds.api.GraphStore graphStore, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)
-
-
Method Detail
-
progressTask
public static org.neo4j.gds.core.utils.progress.tasks.Task progressTask(java.lang.String taskName, LinkPredictionTrainingPipeline pipeline, long relationshipCount)
-
estimate
public static org.neo4j.gds.core.utils.mem.MemoryEstimation estimate(org.neo4j.gds.executor.ExecutionContext executionContext, LinkPredictionTrainingPipeline pipeline, LinkPredictionTrainConfig configuration)
-
generateDatasetSplitGraphFilters
public java.util.Map<PipelineExecutor.DatasetSplits,PipelineGraphFilter> generateDatasetSplitGraphFilters()
-
splitDatasets
public void splitDatasets()
-
execute
protected LinkPredictionTrainPipelineExecutor.LinkPredictionTrainPipelineResult execute(java.util.Map<PipelineExecutor.DatasetSplits,PipelineGraphFilter> dataSplits)
-
getAvailableRelTypesForNodePropertySteps
protected java.util.Set<org.neo4j.gds.RelationshipType> getAvailableRelTypesForNodePropertySteps()
-
additionalGraphStoreCleanup
protected void additionalGraphStoreCleanup(java.util.Map<PipelineExecutor.DatasetSplits,PipelineGraphFilter> datasets)
-
-