Class Nag.Builder

    • Method Detail

      • setLearningRateTracker

        public Nag.Builder setLearningRateTracker​(Tracker learningRateTracker)
        Sets the Tracker for this optimizer.
        Parameters:
        learningRateTracker - the Tracker to be set
        Returns:
        this Builder
      • setMomentum

        public Nag.Builder setMomentum​(float momentum)
        Sets the momentum for Nag.
        Parameters:
        momentum - the value of momentum
        Returns:
        this Builder
      • build

        public Nag build()
        Builds a Nag block.
        Returns:
        the Nag block