Class PolynomialDecayTracker.Builder

  • Enclosing class:
    PolynomialDecayTracker

    public static final class PolynomialDecayTracker.Builder
    extends java.lang.Object
    Builder for PolynomialDecayTracker.
    • Method Detail

      • setBaseValue

        public PolynomialDecayTracker.Builder setBaseValue​(float baseValue)
        Sets the initial value after no steps.
        Parameters:
        baseValue - the initial value
        Returns:
        this Builder
      • setEndLearningRate

        public PolynomialDecayTracker.Builder setEndLearningRate​(float endLearningRate)
        Sets the learning rate at which to end rate decay.
        Parameters:
        endLearningRate - the learning rate at which to end rate decay.
        Returns:
        this builder
      • setDecaySteps

        public PolynomialDecayTracker.Builder setDecaySteps​(int decaySteps)
        Sets the number of training steps to decay learning rate in.
        Parameters:
        decaySteps - the number of training steps to decay learning rate in
        Returns:
        this builder
      • optPower

        public PolynomialDecayTracker.Builder optPower​(float power)
        Sets the power of the polynomial to decay by.
        Parameters:
        power - the power of the polynomial to decay by.
        Returns:
        this builder
      • build

        public PolynomialDecayTracker build()
        Builds a PolynomialDecayTracker.
        Returns:
        a PolynomialDecayTracker