public class AdaGrad extends Object implements Serializable
Modifier and Type | Field and Description |
---|---|
static double |
DEFAULT_ADAGRAD_EPSILON |
INDArray |
historicalGradient |
protected double |
learningRate |
protected int |
numIterations |
long[] |
shape |
Constructor and Description |
---|
AdaGrad(double learningRate) |
AdaGrad(double learningRate,
double epsilon) |
AdaGrad(int rows,
int cols) |
AdaGrad(int rows,
int cols,
double learningRate) |
AdaGrad(long[] shape,
double learningRate) |
Modifier and Type | Method and Description |
---|---|
AdaGrad |
createSubset(int index) |
double |
getGradient(double gradient,
int column,
long[] shape) |
INDArray |
getGradient(INDArray gradient,
int iteration)
Gets feature specific learning rates
Adagrad keeps a history of gradients being passed in.
|
INDArray |
getGradient(INDArray gradient,
int slice,
long[] shape) |
void |
setStateViewArray(INDArray viewArray,
int[] gradientShape,
char gradientOrder,
boolean initialize) |
void |
setStateViewArray(INDArray viewArray,
long[] gradientShape,
char gradientOrder,
boolean initialize) |
int |
stateSizeForInputSize(int inputSize) |
void |
update(Object... args) |
public static final double DEFAULT_ADAGRAD_EPSILON
public INDArray historicalGradient
public long[] shape
protected double learningRate
protected int numIterations
public AdaGrad(int rows, int cols, double learningRate)
rows
- cols
- learningRate
- public AdaGrad(int rows, int cols)
public AdaGrad(long[] shape, double learningRate)
public AdaGrad(double learningRate)
public AdaGrad(double learningRate, double epsilon)
public int stateSizeForInputSize(int inputSize)
public void setStateViewArray(INDArray viewArray, int[] gradientShape, char gradientOrder, boolean initialize)
public void setStateViewArray(INDArray viewArray, long[] gradientShape, char gradientOrder, boolean initialize)
public void update(Object... args)
public INDArray getGradient(INDArray gradient, int iteration)
gradient
- the gradient to get learning rates foriteration
- public double getGradient(double gradient, int column, long[] shape)
public AdaGrad createSubset(int index)
Copyright © 2020. All rights reserved.