Package | Description |
---|---|
org.tribuo.math.optimisers |
Provides implementations of
StochasticGradientOptimiser . |
Modifier and Type | Method and Description |
---|---|
static SGD |
SGD.getLinearDecaySGD(double learningRate)
Generates an SGD optimiser with a linearly decaying learning rate initialised to learningRate.
|
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)
Generates an SGD optimiser with a constant learning rate set to learningRate.
|
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)
Generates an SGD optimiser with a sqrt decaying learning rate initialised to learningRate.
|
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.