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 Summary

Fields

Modifier and Type	Field	Description
int	epochs
DataOptions	general
GradientOptimiserOptions	gradientOptions
int	loggingInterval
TrainTest.LossEnum	loss
int	minibatchSize

Fields inherited from interface com.oracle.labs.mlrg.olcut.config.Options

header

Constructor Summary

Constructors

Constructor	Description
SGDOptions()

Method Summary

Modifier and Type	Method	Description
String	getOptionsDescription()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

general

public DataOptions general

gradientOptions

public GradientOptimiserOptions gradientOptions

epochs

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

loss

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

loggingInterval

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

minibatchSize

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

Constructor Details

SGDOptions

public SGDOptions()

Method Details

getOptionsDescription

public String getOptionsDescription()

Specified by:

getOptionsDescription in interface com.oracle.labs.mlrg.olcut.config.Options