Class LinearSGDOptions

java.lang.Object
org.tribuo.multilabel.sgd.linear.LinearSGDOptions
All Implemented Interfaces:
com.oracle.labs.mlrg.olcut.config.Options

public class LinearSGDOptions extends Object implements com.oracle.labs.mlrg.olcut.config.Options
CLI options for training a linear classifier.
  • Field Details

    • sgoOptions

      public GradientOptimiserOptions sgoOptions
    • sgdEpochs

      @Option(longName="sgd-epochs", usage="Number of SGD epochs.") public int sgdEpochs
    • sgdObjective

      @Option(longName="sgd-objective", usage="Loss function.") public LinearSGDOptions.LossEnum sgdObjective
    • sgdLoggingInterval

      @Option(longName="sgd-logging-interval", usage="Log the objective after <int> examples.") public int sgdLoggingInterval
    • sgdMinibatchSize

      @Option(longName="sgd-minibatch-size", usage="Minibatch size.") public int sgdMinibatchSize
  • Constructor Details

    • LinearSGDOptions

      public LinearSGDOptions()
  • Method Details

    • getLoss

      public MultiLabelObjective getLoss()
      Returns the loss function specified in the arguments.
      Returns:
      The loss function.
    • getTrainer

      public LinearSGDTrainer getTrainer()