Class TrainTest.SGDOptions

java.lang.Object
org.tribuo.regression.sgd.TrainTest.SGDOptions
All Implemented Interfaces:
com.oracle.labs.mlrg.olcut.config.Options
Enclosing class:
TrainTest

public static class TrainTest.SGDOptions extends Object implements com.oracle.labs.mlrg.olcut.config.Options
Command line options.
  • Field Details

    • general

      public DataOptions general
      The dataset loading options.
    • gradientOptions

      public GradientOptimiserOptions gradientOptions
      Gradient descent options.
    • epochs

      @Option(charName='i', longName="epochs", usage="Number of SGD epochs.") public int epochs
      Number of SGD epochs. Defaults to 5.
    • loss

      @Option(charName='o', longName="objective", usage="Loss function.") public TrainTest.LossEnum loss
      Loss function. Defaults to SQUARED.
    • loggingInterval

      @Option(charName='p', longName="logging-interval", usage="Log the objective after <int> examples.") public int loggingInterval
      Log the objective after n examples. Defaults to 100.
    • minibatchSize

      @Option(charName='z', longName="minibatch-size", usage="Minibatch size.") public int minibatchSize
      Minibatch size. Defaults to 1.
  • Constructor Details

    • SGDOptions

      public SGDOptions()
  • Method Details

    • getOptionsDescription

      public String getOptionsDescription()
      Specified by:
      getOptionsDescription in interface com.oracle.labs.mlrg.olcut.config.Options