Package me.yixqiao.jlearn.optimizers
Class Momentum
- java.lang.Object
-
- me.yixqiao.jlearn.optimizers.Optimizer
-
- me.yixqiao.jlearn.optimizers.Momentum
-
public class Momentum extends Optimizer
Gradient descent with momentum.
-
-
Constructor Summary
Constructors Constructor Description Momentum(double learningRate, double momentum)Create a new instance.
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description Matrixapply(Matrix g)Apply the optimizer.OptimizercloneSettings()Return a clone of this optimizer's settings.voidmultiplyLR(double d)Multiply the learning rate.
-
-
-
Method Detail
-
cloneSettings
public Optimizer cloneSettings()
Description copied from class:OptimizerReturn a clone of this optimizer's settings.- Specified by:
cloneSettingsin classOptimizer- Returns:
- the clone
-
multiplyLR
public void multiplyLR(double d)
Description copied from class:OptimizerMultiply the learning rate.Used to scale the learning rate in different batch sizes.
- Specified by:
multiplyLRin classOptimizer- Parameters:
d- amount
-
-