Class LinearSGDOptions
java.lang.Object
org.tribuo.multilabel.sgd.linear.LinearSGDOptions
- All Implemented Interfaces:
com.oracle.labs.mlrg.olcut.config.Options
CLI options for training a linear classifier.
-
Nested Class Summary
Nested Classes -
Field Summary
FieldsModifier and TypeFieldDescriptionint
int
int
Fields inherited from interface com.oracle.labs.mlrg.olcut.config.Options
header
-
Constructor Summary
Constructors -
Method Summary
Modifier and TypeMethodDescriptiongetLoss()
Returns the loss function specified in the arguments.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
getOptionsDescription
-
Field Details
-
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
Returns the loss function specified in the arguments.- Returns:
- The loss function.
-
getTrainer
-