Package com.jml.optimizers

Class Scheduler

java.lang.Object
com.jml.optimizers.Scheduler
Direct Known Subclasses:
StepLearningRate
public abstract class Scheduler extends Object
Learning rate scheduler for optimizers. For many models which are fit using an Optimizer, it may be beneficial to decrease the learning rate according to some rule as the optimizer iterates. This interface provides the ability to define a rule to accomplish this.

  • Constructor Details

    • Scheduler

      public Scheduler()

  • Method Details

    • step

      public abstract void step​(Optimizer optm)
      Applies the specified scheduler rule to the learning rate of the optimizer.
      Parameters:
      optm - Optimizer to apply this Scheduler to.