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

Class LinkPredictionPredictPipelineExecutor

  • public class LinkPredictionPredictPipelineExecutor
extends org.neo4j.gds.ml.pipeline.PipelineExecutor<LinkPredictionPredictPipelineBaseConfig,​org.neo4j.gds.ml.pipeline.linkPipeline.LinkPredictionPipeline,​org.neo4j.gds.ml.linkmodels.LinkPredictionResult>

    • 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

    • Constructor Summary

      Constructors 
      Constructor Description
      LinkPredictionPredictPipelineExecutor​(org.neo4j.gds.ml.pipeline.linkPipeline.LinkPredictionPipeline pipeline, org.neo4j.gds.models.Classifier classifier, LinkPredictionPredictPipelineBaseConfig 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 Static Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      static org.neo4j.gds.core.utils.mem.MemoryEstimation estimate​(org.neo4j.gds.core.model.ModelCatalog modelCatalog, org.neo4j.gds.ml.pipeline.linkPipeline.LinkPredictionPipeline pipeline, LinkPredictionPredictPipelineBaseConfig configuration, int linkFeatureDimension)  
      protected org.neo4j.gds.ml.linkmodels.LinkPredictionResult 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

        cleanUpGraphStore, compute, estimateNodePropertySteps, release, validate, validateTrainingParameterSpace

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

        getProgressTracker, getTerminationFlag, setTerminationFlag

      • Methods inherited from class java.lang.Object

        clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

    • Constructor Detail

      • LinkPredictionPredictPipelineExecutor

        public LinkPredictionPredictPipelineExecutor​(org.neo4j.gds.ml.pipeline.linkPipeline.LinkPredictionPipeline pipeline,
                                             org.neo4j.gds.models.Classifier classifier,
                                             LinkPredictionPredictPipelineBaseConfig 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<LinkPredictionPredictPipelineBaseConfig,​org.neo4j.gds.ml.pipeline.linkPipeline.LinkPredictionPipeline,​org.neo4j.gds.ml.linkmodels.LinkPredictionResult>

      • execute

        protected org.neo4j.gds.ml.linkmodels.LinkPredictionResult 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<LinkPredictionPredictPipelineBaseConfig,​org.neo4j.gds.ml.pipeline.linkPipeline.LinkPredictionPipeline,​org.neo4j.gds.ml.linkmodels.LinkPredictionResult>

      • estimate

        public static org.neo4j.gds.core.utils.mem.MemoryEstimation estimate​(org.neo4j.gds.core.model.ModelCatalog modelCatalog,
                                                                     org.neo4j.gds.ml.pipeline.linkPipeline.LinkPredictionPipeline pipeline,
                                                                     LinkPredictionPredictPipelineBaseConfig configuration,
                                                                     int linkFeatureDimension)