org.deeplearning4j.arbiter.conf.updater

Class AdaGradSpace

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.AdaGradSpace

All Implemented Interfaces:

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

public class AdaGradSpace
extends BaseUpdaterSpace

Constructor Summary

Constructors

Modifier	Constructor and Description
	AdaGradSpace(ParameterSpace<Double> learningRate)
protected	AdaGradSpace(ParameterSpace<Double> learningRate, ParameterSpace<org.nd4j.linalg.schedule.ISchedule> lrSchedule)

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.learning.config.IUpdater	getValue(double[] parameterValues)
static AdaGradSpace	withLR(ParameterSpace<Double> lr)
static AdaGradSpace	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

AdaGradSpace

public AdaGradSpace(ParameterSpace<Double> learningRate)

AdaGradSpace

protected AdaGradSpace(ParameterSpace<Double> learningRate,
                       ParameterSpace<org.nd4j.linalg.schedule.ISchedule> lrSchedule)

Method Detail

withLR

public static AdaGradSpace withLR(ParameterSpace<Double> lr)

withLRSchedule

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

getValue

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