Package org.neo4j.gds.ml.nodemodels

Class NodeClassificationTrainProc

  • public class NodeClassificationTrainProc
extends org.neo4j.gds.TrainProc<org.neo4j.gds.ml.nodemodels.NodeClassificationTrain,​org.neo4j.gds.ml.nodemodels.logisticregression.NodeLogisticRegressionData,​org.neo4j.gds.ml.nodemodels.NodeClassificationTrainConfig,​org.neo4j.gds.ml.nodemodels.NodeClassificationModelInfo>

    • 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.nodemodels.NodeClassificationTrain,​org.neo4j.gds.ml.nodemodels.NodeClassificationTrainConfig> 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.nodemodels.NodeClassificationTrainConfig 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 validateConfigsAfterLoad​(org.neo4j.gds.api.GraphStore graphStore, org.neo4j.gds.config.GraphCreateConfig graphCreateConfig, org.neo4j.gds.ml.nodemodels.NodeClassificationTrainConfig config)  

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

        trainAndStoreModelWithResult, validateConfigsBeforeLoad

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

        algoName, compute, compute, computeEstimate, getOrCreateGraphStore, newConfig, nodeProperties, processInput, sharedConfigKeys, 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

      • NodeClassificationTrainProc

        public NodeClassificationTrainProc()

    • 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)

      • modelType

        protected java.lang.String modelType()
        Specified by:
        modelType in class org.neo4j.gds.TrainProc<org.neo4j.gds.ml.nodemodels.NodeClassificationTrain,​org.neo4j.gds.ml.nodemodels.logisticregression.NodeLogisticRegressionData,​org.neo4j.gds.ml.nodemodels.NodeClassificationTrainConfig,​org.neo4j.gds.ml.nodemodels.NodeClassificationModelInfo>

      • validateConfigsAfterLoad

        protected void validateConfigsAfterLoad​(org.neo4j.gds.api.GraphStore graphStore,
                                        org.neo4j.gds.config.GraphCreateConfig graphCreateConfig,
                                        org.neo4j.gds.ml.nodemodels.NodeClassificationTrainConfig config)
        Overrides:
        validateConfigsAfterLoad in class org.neo4j.gds.AlgoBaseProc<org.neo4j.gds.ml.nodemodels.NodeClassificationTrain,​org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.nodemodels.logisticregression.NodeLogisticRegressionData,​org.neo4j.gds.ml.nodemodels.NodeClassificationTrainConfig,​org.neo4j.gds.ml.nodemodels.NodeClassificationModelInfo>,​org.neo4j.gds.ml.nodemodels.NodeClassificationTrainConfig>

      • newConfig

        protected org.neo4j.gds.ml.nodemodels.NodeClassificationTrainConfig 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.nodemodels.NodeClassificationTrain,​org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.nodemodels.logisticregression.NodeLogisticRegressionData,​org.neo4j.gds.ml.nodemodels.NodeClassificationTrainConfig,​org.neo4j.gds.ml.nodemodels.NodeClassificationModelInfo>,​org.neo4j.gds.ml.nodemodels.NodeClassificationTrainConfig>

      • algorithmFactory

        protected org.neo4j.gds.AlgorithmFactory<org.neo4j.gds.ml.nodemodels.NodeClassificationTrain,​org.neo4j.gds.ml.nodemodels.NodeClassificationTrainConfig> algorithmFactory()
        Specified by:
        algorithmFactory in class org.neo4j.gds.AlgoBaseProc<org.neo4j.gds.ml.nodemodels.NodeClassificationTrain,​org.neo4j.gds.core.model.Model<org.neo4j.gds.ml.nodemodels.logisticregression.NodeLogisticRegressionData,​org.neo4j.gds.ml.nodemodels.NodeClassificationTrainConfig,​org.neo4j.gds.ml.nodemodels.NodeClassificationModelInfo>,​org.neo4j.gds.ml.nodemodels.NodeClassificationTrainConfig>