Class GradientOptimiserOptions

java.lang.Object
org.tribuo.math.optimisers.GradientOptimiserOptions
All Implemented Interfaces:
com.oracle.labs.mlrg.olcut.config.Options

public class GradientOptimiserOptions extends Object implements com.oracle.labs.mlrg.olcut.config.Options
CLI options for configuring a gradient optimiser.
  • Field Details

    • learningRate

      @Option(longName="sgo-learning-rate", usage="Learning rate for AdaGrad, AdaGradRDA, Adam, Pegasos.") public double learningRate
    • epsilon

      @Option(longName="sgo-epsilon", usage="Epsilon for AdaDelta, AdaGrad, AdaGradRDA, Adam.") public double epsilon
    • rho

      @Option(longName="sgo-rho", usage="Rho for RMSProp, AdaDelta, SGD with Momentum.") public double rho
    • lambda

      @Option(longName="sgo-lambda", usage="Lambda for Pegasos.") public double lambda
    • paramAve

      @Option(longName="sgo-parameter-averaging", usage="Use parameter averaging.") public boolean paramAve
    • momentum

      @Option(longName="sgo-momentum", usage="Use momentum in SGD.") public SGD.Momentum momentum
  • Constructor Details

  • Method Details