Package org.tribuo.regression.sgd

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	Number of SGD epochs.
DataOptions	general	The dataset loading options.
GradientOptimiserOptions	gradientOptions	Gradient descent options.
int	loggingInterval	Log the objective after n examples.
TrainTest.LossEnum	loss	Loss function.
int	minibatchSize	Minibatch size.

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

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