public class Nesterovs extends Object implements Serializable, GradientUpdater
Modifier and Type | Class and Description |
---|---|
static class |
Nesterovs.NesterovsAggregator |
Constructor and Description |
---|
Nesterovs(double momentum) |
Nesterovs(double momentum,
double learningRate) |
Modifier and Type | Method and Description |
---|---|
GradientUpdaterAggregator |
getAggregator(boolean addThis)
Get a GradientUpdaterAggregator.
|
INDArray |
getGradient(INDArray gradient,
int iteration)
Get the nesterov update
|
void |
update(Object... args)
update(learningRate,momentum)
|
public Nesterovs(double momentum, double learningRate)
public Nesterovs(double momentum)
public void update(Object... args)
GradientUpdater
update
in interface GradientUpdater
public INDArray getGradient(INDArray gradient, int iteration)
getGradient
in interface GradientUpdater
gradient
- the gradient to get the update foriteration
- public GradientUpdaterAggregator getAggregator(boolean addThis)
GradientUpdater
getAggregator
in interface GradientUpdater
addThis
- If true: return a GradientUpdaterAggregator with the GradientUpdater already added.
If false: return an empty (uninitialized) GradientUpdaterAggregatorCopyright © 2016. All Rights Reserved.