Package | Description |
---|---|
org.tribuo.math.optimisers |
Provides implementations of
StochasticGradientOptimiser . |
Modifier and Type | Field and Description |
---|---|
SGD.Momentum |
GradientOptimiserOptions.momentum |
protected SGD.Momentum |
SGD.useMomentum |
Modifier and Type | Method and Description |
---|---|
static SGD.Momentum |
SGD.Momentum.valueOf(String name)
Returns the enum constant of this type with the specified name.
|
static SGD.Momentum[] |
SGD.Momentum.values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
Modifier and Type | Method and Description |
---|---|
static SGD |
SGD.getLinearDecaySGD(double learningRate,
double rho,
SGD.Momentum momentumType)
Generates an SGD optimiser with a linearly decaying learning rate initialised to learningRate, with momentum.
|
static SGD |
SGD.getSimpleSGD(double learningRate,
double rho,
SGD.Momentum momentumType)
Generates an SGD optimiser with a constant learning rate set to learningRate, with momentum.
|
static SGD |
SGD.getSqrtDecaySGD(double learningRate,
double rho,
SGD.Momentum momentumType)
Generates an SGD optimiser with a sqrt decaying learning rate initialised to learningRate, with momentum.
|
Copyright © 2015–2021 Oracle and/or its affiliates. All rights reserved.