public class Nesterovs extends Object implements Serializable, GradientUpdater
Constructor and Description |
---|
Nesterovs(double momentum)
Initialize with a learning rate of 0.1
|
Nesterovs(double momentum,
double lr) |
Modifier and Type | Method and Description |
---|---|
INDArray |
getGradient(INDArray gradient,
int iteration)
Get the nesterov update
|
double |
getLr() |
double |
getMomentum() |
void |
setLr(double lr) |
void |
setMomentum(double momentum) |
public Nesterovs(double momentum, double lr)
public Nesterovs(double momentum)
momentum
- the momentum to initialize withpublic double getMomentum()
public void setMomentum(double momentum)
public double getLr()
public void setLr(double lr)
public INDArray getGradient(INDArray gradient, int iteration)
getGradient
in interface GradientUpdater
gradient
- the gradient to get the update foriteration
- Copyright © 2015. All Rights Reserved.