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 and 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 and Description
SGDOptions()

Method Summary

Modifier and Type	Method and Description
String	getOptionsDescription()

Methods inherited from class java.lang.Object

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

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

formatUsage, generateTypeDescription, getAllOptions, getEnumConstantString, getOptionFields, getOptions, getOptionUsage, getOptionUsage, getOptionUsage, getUsage

Field Detail

general

public DataOptions general

gradientOptions

public GradientOptimiserOptions gradientOptions

epochs

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

loss

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

loggingInterval

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

minibatchSize

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

Constructor Detail

SGDOptions

public SGDOptions()

Method Detail

getOptionsDescription

public String getOptionsDescription()

Specified by:

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