public class Adam extends Object implements Serializable, GradientUpdater
Constructor and Description |
---|
Adam(double alpha,
double beta1,
double beta2,
double epsilon) |
Modifier and Type | Method and Description |
---|---|
double |
getAlpha() |
double |
getBeta1() |
double |
getBeta2() |
INDArray |
getGradient(INDArray gradient,
int iteration)
Calculate the update based on the given gradient
|
INDArray |
getM() |
INDArray |
getV() |
void |
setAlpha(double alpha) |
void |
setBeta1(double beta1) |
void |
setBeta2(double beta2) |
void |
setM(INDArray m) |
void |
setV(INDArray v) |
public INDArray getGradient(INDArray gradient, int iteration)
getGradient
in interface GradientUpdater
gradient
- the gradient to get the update foriteration
- public double getAlpha()
public void setAlpha(double alpha)
public double getBeta1()
public void setBeta1(double beta1)
public double getBeta2()
public void setBeta2(double beta2)
public INDArray getM()
public void setM(INDArray m)
public INDArray getV()
public void setV(INDArray v)
Copyright © 2015. All Rights Reserved.