Class LinearSGDOptions
java.lang.Object
org.tribuo.classification.sgd.linear.LinearSGDOptions
- All Implemented Interfaces:
com.oracle.labs.mlrg.olcut.config.Options
,ClassificationOptions<LinearSGDTrainer>
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.Constructs the trainer based on the provided 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
-
sgdObjective
@Option(longName="sgd-objective", usage="Loss function. Defaults to LOG.") public LinearSGDOptions.LossEnum sgdObjective -
sgdLoggingInterval
@Option(longName="sgd-logging-interval", usage="Log the objective after <int> examples. Defaults to 100.") public int sgdLoggingInterval -
sgdMinibatchSize
@Option(longName="sgd-minibatch-size", usage="Minibatch size. Defaults to 1.") public int sgdMinibatchSize
-
-
Constructor Details
-
LinearSGDOptions
public LinearSGDOptions()
-
-
Method Details
-
getLoss
Returns the loss function specified in the arguments.- Returns:
- The loss function.
-
getTrainer
Description copied from interface:ClassificationOptions
Constructs the trainer based on the provided arguments.- Specified by:
getTrainer
in interfaceClassificationOptions<LinearSGDTrainer>
- Returns:
- The trainer.
-