Package org.neo4j.gds.ml.gradientdescent

Class Training

java.lang.Object
org.neo4j.gds.ml.gradientdescent.Training
public class Training extends Object

  • Constructor Summary

    Constructors
    Constructor
    Description
    Training(GradientDescentConfig config, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker, org.neo4j.gds.core.utils.progress.tasks.LogLevel messageLogLevel, long trainSize, org.neo4j.gds.termination.TerminationFlag terminationFlag)
     

  • Method Summary

    Modifier and Type
    Method
    Description
    static org.neo4j.gds.core.utils.mem.MemoryEstimation
    memoryEstimation(int numberOfFeatures, int numberOfClasses)
     
    static org.neo4j.gds.core.utils.mem.MemoryEstimation
    memoryEstimation(org.neo4j.gds.core.utils.mem.MemoryRange numberOfFeaturesRange, int numberOfClasses)
     
    void
    train(Objective<?> objective, Supplier<org.neo4j.gds.ml.core.batch.BatchQueue> queueSupplier, int concurrency)
     

    Methods inherited from class java.lang.Object

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

  • Constructor Details

    • Training

      public Training(GradientDescentConfig config, org.neo4j.gds.core.utils.progress.tasks.ProgressTracker progressTracker, org.neo4j.gds.core.utils.progress.tasks.LogLevel messageLogLevel, long trainSize, org.neo4j.gds.termination.TerminationFlag terminationFlag)

  • Method Details

    • memoryEstimation

      public static org.neo4j.gds.core.utils.mem.MemoryEstimation memoryEstimation(int numberOfFeatures, int numberOfClasses)

    • memoryEstimation

      public static org.neo4j.gds.core.utils.mem.MemoryEstimation memoryEstimation(org.neo4j.gds.core.utils.mem.MemoryRange numberOfFeaturesRange, int numberOfClasses)

    • train

      public void train(Objective<?> objective, Supplier<org.neo4j.gds.ml.core.batch.BatchQueue> queueSupplier, int concurrency)