public class RmsPropUpdater extends Object implements GradientUpdater
Constructor and Description |
---|
RmsPropUpdater(double lr,
double rmsDecay) |
Modifier and Type | Method and Description |
---|---|
INDArray |
getGradient(INDArray gradient,
int iteration)
Modify the gradient
to be an update
|
double |
getLR() |
double |
getRmsDecay() |
void |
setLR(double lr) |
void |
setRmsDecay(double rmsDecay) |
public void setRmsDecay(double rmsDecay)
public double getRmsDecay()
public void setLR(double lr)
public double getLR()
public INDArray getGradient(INDArray gradient, int iteration)
GradientUpdater
getGradient
in interface GradientUpdater
gradient
- the gradient to modifyCopyright © 2015. All Rights Reserved.