Uses of Enum Class
org.tribuo.math.optimisers.SGD.Momentum
-
Uses of SGD.Momentum in org.tribuo.math.optimisers
Modifier and TypeFieldDescriptionGradientOptimiserOptions.momentum
Use momentum in SGD.protected SGD.Momentum
SGD.useMomentum
Modifier and TypeMethodDescriptionstatic SGD.Momentum
Returns the enum constant of this class with the specified name.static SGD.Momentum[]
SGD.Momentum.values()
Returns an array containing the constants of this enum class, in the order they are declared.Modifier and TypeMethodDescriptionstatic 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.