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

Class LinkPredictionTrainPipelineExecutor

  • All Implemented Interfaces:
    org.neo4j.gds.core.utils.TerminationFlag
    public class LinkPredictionTrainPipelineExecutor
extends org.neo4j.gds.ml.pipeline.PipelineExecutor<LinkPredictionTrainConfig,​org.neo4j.gds.ml.pipeline.linkPipeline.LinkPredictionPipeline,​LinkPredictionTrainResult>

    • Nested Class Summary

      • Nested classes/interfaces inherited from class org.neo4j.gds.ml.pipeline.PipelineExecutor

        org.neo4j.gds.ml.pipeline.PipelineExecutor.DatasetSplits, org.neo4j.gds.ml.pipeline.PipelineExecutor.GraphFilter

    • Field Summary

      • Fields inherited from class org.neo4j.gds.ml.pipeline.PipelineExecutor

        config, executionContext, graphName, graphStore, pipeline

      • Fields inherited from class org.neo4j.gds.Algorithm

        progressTracker, terminationFlag

      • Fields inherited from interface org.neo4j.gds.core.utils.TerminationFlag

        RUN_CHECK_NODE_COUNT, RUNNING_TRUE

    • Constructor Summary

      Constructors 
      Constructor Description
      LinkPredictionTrainPipelineExecutor​(org.neo4j.gds.ml.pipeline.linkPipeline.LinkPredictionPipeline pipeline, LinkPredictionTrainConfig config, org.neo4j.gds.executor.ExecutionContext executionContext, org.neo4j.gds.api.GraphStore graphStore, java.lang.String graphName, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)  

    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      protected void cleanUpGraphStore​(java.util.Map<org.neo4j.gds.ml.pipeline.PipelineExecutor.DatasetSplits,​org.neo4j.gds.ml.pipeline.PipelineExecutor.GraphFilter> datasets)  
      protected LinkPredictionTrainResult execute​(java.util.Map<org.neo4j.gds.ml.pipeline.PipelineExecutor.DatasetSplits,​org.neo4j.gds.ml.pipeline.PipelineExecutor.GraphFilter> dataSplits)  
      java.util.Map<org.neo4j.gds.ml.pipeline.PipelineExecutor.DatasetSplits,​org.neo4j.gds.ml.pipeline.PipelineExecutor.GraphFilter> splitDataset()  

      • Methods inherited from class org.neo4j.gds.ml.pipeline.PipelineExecutor

        compute, release, validate

      • Methods inherited from class org.neo4j.gds.Algorithm

        getProgressTracker, getTerminationFlag, running, setTerminationFlag

      • 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.core.utils.TerminationFlag

        assertRunning

    • Constructor Detail

      • LinkPredictionTrainPipelineExecutor

        public LinkPredictionTrainPipelineExecutor​(org.neo4j.gds.ml.pipeline.linkPipeline.LinkPredictionPipeline pipeline,
                                           LinkPredictionTrainConfig config,
                                           org.neo4j.gds.executor.ExecutionContext executionContext,
                                           org.neo4j.gds.api.GraphStore graphStore,
                                           java.lang.String graphName,
                                           org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker)

    • Method Detail

      • splitDataset

        public java.util.Map<org.neo4j.gds.ml.pipeline.PipelineExecutor.DatasetSplits,​org.neo4j.gds.ml.pipeline.PipelineExecutor.GraphFilter> splitDataset()
        Specified by:
        splitDataset in class org.neo4j.gds.ml.pipeline.PipelineExecutor<LinkPredictionTrainConfig,​org.neo4j.gds.ml.pipeline.linkPipeline.LinkPredictionPipeline,​LinkPredictionTrainResult>

      • execute

        protected LinkPredictionTrainResult execute​(java.util.Map<org.neo4j.gds.ml.pipeline.PipelineExecutor.DatasetSplits,​org.neo4j.gds.ml.pipeline.PipelineExecutor.GraphFilter> dataSplits)
        Specified by:
        execute in class org.neo4j.gds.ml.pipeline.PipelineExecutor<LinkPredictionTrainConfig,​org.neo4j.gds.ml.pipeline.linkPipeline.LinkPredictionPipeline,​LinkPredictionTrainResult>

      • cleanUpGraphStore

        protected void cleanUpGraphStore​(java.util.Map<org.neo4j.gds.ml.pipeline.PipelineExecutor.DatasetSplits,​org.neo4j.gds.ml.pipeline.PipelineExecutor.GraphFilter> datasets)
        Overrides:
        cleanUpGraphStore in class org.neo4j.gds.ml.pipeline.PipelineExecutor<LinkPredictionTrainConfig,​org.neo4j.gds.ml.pipeline.linkPipeline.LinkPredictionPipeline,​LinkPredictionTrainResult>