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

Class LogisticRegressionTrainer

java.lang.Object
org.neo4j.gds.ml.models.logisticregression.LogisticRegressionTrainer
All Implemented Interfaces:
ClassifierTrainer
public final class LogisticRegressionTrainer extends Object implements ClassifierTrainer

  • Constructor Summary

    Constructors
    Constructor
    Description
    LogisticRegressionTrainer(int concurrency, LogisticRegressionTrainConfig trainConfig, int numberOfClasses, boolean reduceClassCount, org.neo4j.gds.termination.TerminationFlag terminationFlag, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker, org.neo4j.gds.core.utils.progress.tasks.LogLevel messageLogLevel)
     

  • Method Summary

    Modifier and Type
    Method
    Description
    static org.neo4j.gds.core.utils.mem.MemoryEstimation
    memoryEstimation(boolean isReduced, int numberOfClasses, org.neo4j.gds.core.utils.mem.MemoryRange featureDimension, int batchSize, LongUnaryOperator numberOfTrainingExamples)
     
    LogisticRegressionClassifier
    train(Features features, org.neo4j.gds.collections.ha.HugeIntArray labels, org.neo4j.gds.core.utils.paged.ReadOnlyHugeLongArray trainSet)
     

    Methods inherited from class java.lang.Object

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

  • Constructor Details

    • LogisticRegressionTrainer

      public LogisticRegressionTrainer(int concurrency, LogisticRegressionTrainConfig trainConfig, int numberOfClasses, boolean reduceClassCount, org.neo4j.gds.termination.TerminationFlag terminationFlag, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker, org.neo4j.gds.core.utils.progress.tasks.LogLevel messageLogLevel)

  • Method Details

    • memoryEstimation

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

    • train

      public LogisticRegressionClassifier train(Features features, org.neo4j.gds.collections.ha.HugeIntArray labels, org.neo4j.gds.core.utils.paged.ReadOnlyHugeLongArray trainSet)
      Specified by:
      train in interface ClassifierTrainer