Class GradientOptimiserOptions
java.lang.Object
org.tribuo.math.optimisers.GradientOptimiserOptions
- All Implemented Interfaces:
com.oracle.labs.mlrg.olcut.config.Options
public class GradientOptimiserOptions
extends Object
implements com.oracle.labs.mlrg.olcut.config.Options
CLI options for configuring a gradient optimiser.
-
Nested Class Summary
Nested ClassesModifier and TypeClassDescriptionstatic enum
Type of the gradient optimisers available in CLIs. -
Field Summary
FieldsModifier and TypeFieldDescriptiondouble
double
double
boolean
double
Fields inherited from interface com.oracle.labs.mlrg.olcut.config.Options
header
-
Constructor Summary
Constructors -
Method Summary
Modifier and TypeMethodDescriptionGets the configured gradient optimiser.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
-
learningRate
@Option(longName="sgo-learning-rate", usage="Learning rate for AdaGrad, AdaGradRDA, Adam, Pegasos.") public double learningRate -
epsilon
@Option(longName="sgo-epsilon", usage="Epsilon for AdaDelta, AdaGrad, AdaGradRDA, Adam.") public double epsilon -
rho
@Option(longName="sgo-rho", usage="Rho for RMSProp, AdaDelta, SGD with Momentum.") public double rho -
lambda
-
paramAve
@Option(longName="sgo-parameter-averaging", usage="Use parameter averaging.") public boolean paramAve -
momentum
-
-
Constructor Details
-
GradientOptimiserOptions
public GradientOptimiserOptions()
-
-
Method Details
-
getOptimiser
Gets the configured gradient optimiser.- Returns:
- The gradient optimiser.
-