public class AdaDelta extends Object implements Serializable, GradientUpdater
Modifier and Type | Method and Description |
---|---|
INDArray |
getGradient(INDArray gradient,
int iteration)
Get the updated gradient for the given gradient
and also update the state of ada delta.
|
INDArray |
getMsdx() |
INDArray |
getMsg() |
double |
getRho() |
void |
setMsdx(INDArray msdx) |
void |
setMsg(INDArray msg) |
void |
setRho(double rho) |
public INDArray getMsg()
public void setMsg(INDArray msg)
public INDArray getMsdx()
public void setMsdx(INDArray msdx)
public double getRho()
public void setRho(double rho)
public INDArray getGradient(INDArray gradient, int iteration)
getGradient
in interface GradientUpdater
gradient
- the gradient to get the
updated gradient foriteration
- Copyright © 2015. All Rights Reserved.