org.nd4j.linalg.learning

Interface GradientUpdater

All Superinterfaces:

Serializable

All Known Implementing Classes:

AdaDelta, AdaGrad, Adam, Nesterovs, RmsPropUpdater, SgdUpdater

public interface GradientUpdater
extends Serializable

Gradient modifications: Calculates an update and tracks related information for gradient changes over time for handling updates.

Author:

Adam Gibson

Method Summary

Modifier and Type	Method and Description
INDArray	getGradient(INDArray gradient, int iteration) Modify the gradient to be an update

Method Detail

getGradient

INDArray getGradient(INDArray gradient,
                     int iteration)

Modify the gradient to be an update

Parameters:

gradient - the gradient to modify

iteration -

Returns:

the modified gradient