org.tribuo.math.optimisers

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 Classes

Modifier and Type	Class and Description
static class	GradientOptimiserOptions.StochasticGradientOptimiserType Type of the gradient optimisers available in CLIs.

Field Summary

Fields

Modifier and Type	Field and Description
double	epsilon
double	lambda
double	learningRate
SGD.Momentum	momentum
boolean	paramAve
double	rho

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

header

Constructor Summary

Constructors

Constructor and Description
GradientOptimiserOptions()

Method Summary

Modifier and Type	Method and Description
StochasticGradientOptimiser	getOptimiser() Gets 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

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

Field Detail

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

@Option(longName="sgo-lambda",
        usage="Lambda for Pegasos.")
public double lambda

paramAve

@Option(longName="sgo-parameter-averaging",
        usage="Use parameter averaging.")
public boolean paramAve

momentum

@Option(longName="sgo-momentum",
        usage="Use momentum in SGD.")
public SGD.Momentum momentum

Constructor Detail

GradientOptimiserOptions

public GradientOptimiserOptions()

Method Detail

getOptimiser

public StochasticGradientOptimiser getOptimiser()

Gets the configured gradient optimiser.

Returns:

The gradient optimiser.