org.nd4j.linalg.learning

Class NadamUpdater

java.lang.Object

org.nd4j.linalg.learning.NadamUpdater

All Implemented Interfaces:

GradientUpdater<Nadam>

public class NadamUpdater
extends Object
implements GradientUpdater<Nadam>

The Nadam updater. https://arxiv.org/pdf/1609.04747.pdf

Author:

Andrey Spiridonov

Constructor Summary

Constructors

Constructor and Description
NadamUpdater(Nadam config)

Method Summary

Modifier and Type	Method and Description
void	applyUpdater(INDArray gradient, int iteration) Calculate the update based on the given gradient
void	setStateViewArray(INDArray viewArray, int[] gradientShape, char gradientOrder, boolean initialize) For the internal updater state (if any): set this to use the provided array.

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.nd4j.linalg.learning.GradientUpdater

getConfig

Constructor Detail

NadamUpdater

public NadamUpdater(Nadam config)

Method Detail

setStateViewArray

public void setStateViewArray(INDArray viewArray,
                              int[] gradientShape,
                              char gradientOrder,
                              boolean initialize)

Description copied from interface: GradientUpdater

For the internal updater state (if any): set this to use the provided array. Used during initialization, and when restoring the updater state (after serialization, for example)

Specified by:

setStateViewArray in interface GradientUpdater<Nadam>

Parameters:

viewArray - Array (that is a view of a larger array) to use for the state.

initialize - If true: the updater must initialize the view array. If false: no change to view array contents

applyUpdater

public void applyUpdater(INDArray gradient,
                         int iteration)

Calculate the update based on the given gradient

Specified by:

applyUpdater in interface GradientUpdater<Nadam>

Parameters:

gradient - the gradient to get the update for

iteration -