Package org.neo4j.gds.ml.linkmodels

Class LinkPredictionTrainProc

  • public class LinkPredictionTrainProc
extends org.neo4j.gds.TrainProc<org.neo4j.gds.ml.linkmodels.LinkPredictionTrain,​org.neo4j.gds.ml.linkmodels.logisticregression.LinkLogisticRegressionData,​org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig,​org.neo4j.gds.ml.linkmodels.LinkPredictionModelInfo>

    • Nested Class Summary

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

        org.neo4j.gds.TrainProc.TrainResult

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

        org.neo4j.gds.AlgoBaseProc.ComputationResult<A extends org.neo4j.gds.Algorithm<A,​RESULT>,​RESULT extends java.lang.Object,​CONFIG extends org.neo4j.gds.config.AlgoBaseConfig>

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

        org.neo4j.gds.BaseProc.FreeMemoryInspector

    • Field Summary

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

        STATS_DESCRIPTION

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

        allocationTracker, api, callContext, ESTIMATE_DESCRIPTION, log, procedureTransaction, taskRegistryFactory, transaction

    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      protected org.neo4j.gds.AlgorithmFactory<org.neo4j.gds.ml.linkmodels.LinkPredictionTrain,​org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig> algorithmFactory()  
      java.util.stream.Stream<org.neo4j.gds.results.MemoryEstimateResult> estimate​(java.lang.Object graphNameOrConfig, java.util.Map<java.lang.String,​java.lang.Object> configuration)  
      protected java.lang.String modelType()  
      protected org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig newConfig​(java.lang.String username, java.util.Optional<java.lang.String> graphName, java.util.Optional<org.neo4j.gds.config.GraphCreateConfig> maybeImplicitCreate, org.neo4j.gds.core.CypherMapWrapper config)  
      java.util.stream.Stream<MLTrainResult> train​(java.lang.Object graphNameOrConfig, java.util.Map<java.lang.String,​java.lang.Object> configuration)  
      protected void validateConfigsBeforeLoad​(org.neo4j.gds.config.GraphCreateConfig graphCreateConfig, org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig config)  

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

        trainAndStoreModelWithResult

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

        algoName, compute, compute, computeEstimate, getOrCreateGraphStore, newConfig, nodeProperties, processInput, sharedConfigKeys, validateConfigsAfterLoad, validateConfigWithGraphStore

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

        allocationTracker, catalogRequest, databaseId, estimateGraphCreate, graphStoreFromCatalog, isGdsAdmin, newLoader, runWithExceptionLogging, runWithExceptionLogging, tryValidateMemoryUsage, tryValidateMemoryUsage, username, validateConfig, validateConfig, validateGraphName

      • Methods inherited from class java.lang.Object

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

    • Constructor Detail

      • LinkPredictionTrainProc

        public LinkPredictionTrainProc()

    • Method Detail

      • train

        public java.util.stream.Stream<MLTrainResult> train​(java.lang.Object graphNameOrConfig,
                                                    java.util.Map<java.lang.String,​java.lang.Object> configuration)

      • estimate

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

      • newConfig

        protected org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig newConfig​(java.lang.String username,
                                                                          java.util.Optional<java.lang.String> graphName,
                                                                          java.util.Optional<org.neo4j.gds.config.GraphCreateConfig> maybeImplicitCreate,
                                                                          org.neo4j.gds.core.CypherMapWrapper config)
        Specified by:
        newConfig in class org.neo4j.gds.AlgoBaseProc<org.neo4j.gds.ml.linkmodels.LinkPredictionTrain,​org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.linkmodels.logisticregression.LinkLogisticRegressionData,​org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig,​org.neo4j.gds.ml.linkmodels.LinkPredictionModelInfo>,​org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig>

      • modelType

        protected java.lang.String modelType()
        Specified by:
        modelType in class org.neo4j.gds.TrainProc<org.neo4j.gds.ml.linkmodels.LinkPredictionTrain,​org.neo4j.gds.ml.linkmodels.logisticregression.LinkLogisticRegressionData,​org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig,​org.neo4j.gds.ml.linkmodels.LinkPredictionModelInfo>

      • validateConfigsBeforeLoad

        protected void validateConfigsBeforeLoad​(org.neo4j.gds.config.GraphCreateConfig graphCreateConfig,
                                         org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig config)
        Overrides:
        validateConfigsBeforeLoad in class org.neo4j.gds.TrainProc<org.neo4j.gds.ml.linkmodels.LinkPredictionTrain,​org.neo4j.gds.ml.linkmodels.logisticregression.LinkLogisticRegressionData,​org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig,​org.neo4j.gds.ml.linkmodels.LinkPredictionModelInfo>

      • algorithmFactory

        protected org.neo4j.gds.AlgorithmFactory<org.neo4j.gds.ml.linkmodels.LinkPredictionTrain,​org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig> algorithmFactory()
        Specified by:
        algorithmFactory in class org.neo4j.gds.AlgoBaseProc<org.neo4j.gds.ml.linkmodels.LinkPredictionTrain,​org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.linkmodels.logisticregression.LinkLogisticRegressionData,​org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig,​org.neo4j.gds.ml.linkmodels.LinkPredictionModelInfo>,​org.neo4j.gds.ml.linkmodels.LinkPredictionTrainConfig>