Class Sgd.Builder

    • Method Detail

      • setLearningRateTracker

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

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

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