Package org.neo4j.gds.ml.models.logisticregression

Interface LogisticRegressionData

All Superinterfaces:
BaseModelData, Classifier.ClassifierData, Serializable
public interface LogisticRegressionData extends Classifier.ClassifierData, Serializable

  • Method Summary

    Modifier and Type
    Method
    Description
    org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Vector>
    bias()
     
    static org.neo4j.gds.ml.models.logisticregression.ImmutableLogisticRegressionData.Builder
    builder()
     
    default int
    featureDimension()
     
    static org.neo4j.gds.core.utils.mem.MemoryEstimation
    memoryEstimation(boolean isReduced, int numberOfClasses, org.neo4j.gds.core.utils.mem.MemoryRange featureDimension)
     
    static LogisticRegressionData
    standard(int featureCount, int numberOfClasses)
     
    default org.neo4j.gds.ml.api.TrainingMethod
    trainerMethod()
     
    org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Matrix>
    weights()
     
    static LogisticRegressionData
    withReducedClassCount(int featureCount, int numberOfClasses)
     

    Methods inherited from interface org.neo4j.gds.ml.models.Classifier.ClassifierData

    numberOfClasses

  • Method Details

    • weights

      org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Matrix> weights()

    • bias

      org.neo4j.gds.ml.core.functions.Weights<org.neo4j.gds.ml.core.tensor.Vector> bias()

    • trainerMethod

      @Derived default org.neo4j.gds.ml.api.TrainingMethod trainerMethod()
      Specified by:
      trainerMethod in interface BaseModelData

    • featureDimension

      @Derived default int featureDimension()
      Specified by:
      featureDimension in interface BaseModelData

    • standard

      static LogisticRegressionData standard(int featureCount, int numberOfClasses)

    • withReducedClassCount

      static LogisticRegressionData withReducedClassCount(int featureCount, int numberOfClasses)

    • memoryEstimation

      static org.neo4j.gds.core.utils.mem.MemoryEstimation memoryEstimation(boolean isReduced, int numberOfClasses, org.neo4j.gds.core.utils.mem.MemoryRange featureDimension)

    • builder

      static org.neo4j.gds.ml.models.logisticregression.ImmutableLogisticRegressionData.Builder builder()