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

Class LinkPredictionPipelineTrainProc

    • Nested Class Summary

      Nested Classes 
      Modifier and Type Class Description
      static class  LinkPredictionPipelineTrainProc.LPTrainResult  

      • Nested classes/interfaces inherited from class org.neo4j.gds.TrainProc

        org.neo4j.gds.TrainProc.TrainingConfigValidation<TRAIN_CONFIG extends org.neo4j.gds.model.ModelConfig & org.neo4j.gds.config.AlgoBaseConfig>, org.neo4j.gds.TrainProc.TrainResult

    • Field Summary

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

        STATS_DESCRIPTION

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

        api, callContext, ESTIMATE_DESCRIPTION, internalModelCatalog, log, procedureTransaction, taskRegistryFactory, transaction, userLogRegistryFactory, username

    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      org.neo4j.gds.GraphStoreAlgorithmFactory<LinkPredictionTrainPipelineExecutor,​org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainConfig> algorithmFactory()  
      protected LinkPredictionPipelineTrainProc.LPTrainResult constructProcResult​(org.neo4j.gds.executor.ComputationResult<LinkPredictionTrainPipelineExecutor,​org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainResult,​org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainConfig> computationResult)  
      java.util.stream.Stream<org.neo4j.gds.results.MemoryEstimateResult> estimate​(java.lang.Object graphNameOrConfiguration, java.util.Map<java.lang.String,​java.lang.Object> algoConfiguration)  
      protected org.neo4j.gds.core.model.Model<?,​?,​?> extractModel​(org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainResult algoResult)  
      protected java.lang.String modelType()  
      protected org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainConfig newConfig​(java.lang.String username, org.neo4j.gds.core.CypherMapWrapper config)  
      java.util.stream.Stream<LinkPredictionPipelineTrainProc.LPTrainResult> train​(java.lang.String graphName, java.util.Map<java.lang.String,​java.lang.Object> config)  

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

        computationResultConsumer, trainAndStoreModelWithResult, validationConfig, withModelCatalog

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

        compute, compute, computeEstimate, configParser, name, newConfigFunction, nodeProperties

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

        databaseId, executionContext, graphLoaderContext, graphStoreFromCatalog, isGdsAdmin, memoryUsageValidator, modelCatalog, runWithExceptionLogging, runWithExceptionLogging, setModelCatalog, username, validateConfig, validateConfig, validateGraphName

      • 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.executor.AlgorithmSpec

        createDefaultExecutorSpec

    • Constructor Detail

      • LinkPredictionPipelineTrainProc

        public LinkPredictionPipelineTrainProc()

    • Method Detail

      • estimate

        public java.util.stream.Stream<org.neo4j.gds.results.MemoryEstimateResult> estimate​(java.lang.Object graphNameOrConfiguration,
                                                                                    java.util.Map<java.lang.String,​java.lang.Object> algoConfiguration)

      • newConfig

        protected org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainConfig newConfig​(java.lang.String username,
                                                                                           org.neo4j.gds.core.CypherMapWrapper config)
        Specified by:
        newConfig in class org.neo4j.gds.AlgoBaseProc<LinkPredictionTrainPipelineExecutor,​org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainResult,​org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainConfig,​LinkPredictionPipelineTrainProc.LPTrainResult>

      • algorithmFactory

        public org.neo4j.gds.GraphStoreAlgorithmFactory<LinkPredictionTrainPipelineExecutor,​org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainConfig> algorithmFactory()

      • extractModel

        protected org.neo4j.gds.core.model.Model<?,​?,​?> extractModel​(org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainResult algoResult)
        Specified by:
        extractModel in class org.neo4j.gds.TrainProc<LinkPredictionTrainPipelineExecutor,​org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainResult,​org.neo4j.gds.ml.pipeline.linkPipeline.train.LinkPredictionTrainConfig,​LinkPredictionPipelineTrainProc.LPTrainResult>