org.nd4j.linalg.learning

Class SgdUpdater

java.lang.Object

org.nd4j.linalg.learning.SgdUpdater

All Implemented Interfaces:

Serializable, GradientUpdater

public class SgdUpdater
extends Object
implements GradientUpdater

Author:

Adam Gibson

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
SgdUpdater(double lr)

Method Summary

Modifier and Type	Method and Description
INDArray	getGradient(INDArray gradient, int iteration) Modify the gradient to be an update
double	getLr()
void	setLr(double lr)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

SgdUpdater

public SgdUpdater(double lr)

Method Detail

getLr

public double getLr()

setLr

public void setLr(double lr)

getGradient

public INDArray getGradient(INDArray gradient,
                            int iteration)

Description copied from interface: GradientUpdater

Modify the gradient to be an update

Specified by:

getGradient in interface GradientUpdater

Parameters:

gradient - the gradient to modify

Returns:

the modified gradient