Package 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	Description
static enum	GradientOptimiserOptions.StochasticGradientOptimiserType	Type of the gradient optimisers available in CLIs.

Field Summary

Fields

Modifier and Type	Field	Description
double	epsilon	Epsilon for AdaDelta, AdaGrad, AdaGradRDA, Adam.
double	lambda	Lambda for Pegasos.
double	learningRate	Learning rate for AdaGrad, AdaGradRDA, Adam, Pegasos.
SGD.Momentum	momentum	Use momentum in SGD.
boolean	paramAve	Use parameter averaging.
double	rho	Rho for RMSProp, AdaDelta, SGD with Momentum.

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

header

Constructor Summary

Constructors

Constructor	Description
GradientOptimiserOptions()

Method Summary

Modifier and Type	Method	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

getOptionsDescription

Field Details

learningRate

@Option(longName="sgo-learning-rate",
        usage="Learning rate for AdaGrad, AdaGradRDA, Adam, Pegasos.")
public double learningRate

Learning rate for AdaGrad, AdaGradRDA, Adam, Pegasos.

epsilon

@Option(longName="sgo-epsilon",
        usage="Epsilon for AdaDelta, AdaGrad, AdaGradRDA, Adam.")
public double epsilon

Epsilon for AdaDelta, AdaGrad, AdaGradRDA, Adam.

rho

@Option(longName="sgo-rho",
        usage="Rho for RMSProp, AdaDelta, SGD with Momentum.")
public double rho

Rho for RMSProp, AdaDelta, SGD with Momentum.

lambda

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

Lambda for Pegasos.

paramAve

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

Use parameter averaging.

momentum

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

Use momentum in SGD.

Constructor Details

GradientOptimiserOptions

public GradientOptimiserOptions()

Method Details

getOptimiser

public StochasticGradientOptimiser getOptimiser()

Gets the configured gradient optimiser.

Returns:

The gradient optimiser.