org.deeplearning4j.arbiter.conf.updater

Class NadamSpace

java.lang.Object

org.deeplearning4j.arbiter.optimize.api.AbstractParameterSpace<org.nd4j.linalg.learning.config.IUpdater>

org.deeplearning4j.arbiter.conf.updater.BaseUpdaterSpace

org.deeplearning4j.arbiter.conf.updater.NadamSpace

All Implemented Interfaces:

ParameterSpace<org.nd4j.linalg.learning.config.IUpdater>

public class NadamSpace
extends BaseUpdaterSpace

Constructor Summary

Constructors

Constructor and Description
NadamSpace(ParameterSpace<Double> learningRate)
NadamSpace(ParameterSpace<Double> learningRate, ParameterSpace<Double> beta1, ParameterSpace<Double> beta2, ParameterSpace<Double> epsilon)
NadamSpace(ParameterSpace<Double> learningRate, ParameterSpace<org.nd4j.linalg.schedule.ISchedule> learningRateSchedule, ParameterSpace<Double> beta1, ParameterSpace<Double> beta2, ParameterSpace<Double> epsilon)

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.learning.config.IUpdater	getValue(double[] parameterValues)
static NadamSpace	withLR(ParameterSpace<Double> lr)
static NadamSpace	withLRSchedule(ParameterSpace<org.nd4j.linalg.schedule.ISchedule> lrSchedule)

Methods inherited from class org.deeplearning4j.arbiter.conf.updater.BaseUpdaterSpace

collectLeaves, isLeaf, numParameters, setIndices

Methods inherited from class org.deeplearning4j.arbiter.optimize.api.AbstractParameterSpace

getNestedSpaces

Methods inherited from class java.lang.Object

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

Constructor Detail

NadamSpace

public NadamSpace(ParameterSpace<Double> learningRate)

NadamSpace

public NadamSpace(ParameterSpace<Double> learningRate,
                  ParameterSpace<Double> beta1,
                  ParameterSpace<Double> beta2,
                  ParameterSpace<Double> epsilon)

NadamSpace

public NadamSpace(ParameterSpace<Double> learningRate,
                  ParameterSpace<org.nd4j.linalg.schedule.ISchedule> learningRateSchedule,
                  ParameterSpace<Double> beta1,
                  ParameterSpace<Double> beta2,
                  ParameterSpace<Double> epsilon)

Method Detail

withLR

public static NadamSpace withLR(ParameterSpace<Double> lr)

withLRSchedule

public static NadamSpace withLRSchedule(ParameterSpace<org.nd4j.linalg.schedule.ISchedule> lrSchedule)

getValue

public org.nd4j.linalg.learning.config.IUpdater getValue(double[] parameterValues)