org.deeplearning4j.arbiter.conf.updater.schedule

Class StepScheduleSpace

java.lang.Object

org.deeplearning4j.arbiter.conf.updater.schedule.StepScheduleSpace

All Implemented Interfaces:

ParameterSpace<org.nd4j.linalg.schedule.ISchedule>

public class StepScheduleSpace
extends Object
implements ParameterSpace<org.nd4j.linalg.schedule.ISchedule>

Constructor Summary

Constructors

Constructor and Description
StepScheduleSpace(@NonNull org.nd4j.linalg.schedule.ScheduleType scheduleType, @NonNull ParameterSpace<Double> initialValue, double decayRate, double step)
StepScheduleSpace(@NonNull org.nd4j.linalg.schedule.ScheduleType scheduleType, @NonNull ParameterSpace<Double> initialValue, @NonNull ParameterSpace<Double> decayRate, @NonNull ParameterSpace<Double> step)

Method Summary

Modifier and Type	Method and Description
List<ParameterSpace>	collectLeaves()
Map<String,ParameterSpace>	getNestedSpaces()
org.nd4j.linalg.schedule.ISchedule	getValue(double[] parameterValues)
boolean	isLeaf()
int	numParameters()
void	setIndices(int... indices)

Methods inherited from class java.lang.Object

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

Constructor Detail

StepScheduleSpace

public StepScheduleSpace(@NonNull
                         @NonNull org.nd4j.linalg.schedule.ScheduleType scheduleType,
                         @NonNull
                         @NonNull ParameterSpace<Double> initialValue,
                         double decayRate,
                         double step)

StepScheduleSpace

public StepScheduleSpace(@NonNull
                         @NonNull org.nd4j.linalg.schedule.ScheduleType scheduleType,
                         @NonNull
                         @NonNull ParameterSpace<Double> initialValue,
                         @NonNull
                         @NonNull ParameterSpace<Double> decayRate,
                         @NonNull
                         @NonNull ParameterSpace<Double> step)

Method Detail

getValue

public org.nd4j.linalg.schedule.ISchedule getValue(double[] parameterValues)

Specified by:

getValue in interface ParameterSpace<org.nd4j.linalg.schedule.ISchedule>

numParameters

public int numParameters()

Specified by:

numParameters in interface ParameterSpace<org.nd4j.linalg.schedule.ISchedule>

collectLeaves

public List<ParameterSpace> collectLeaves()

Specified by:

collectLeaves in interface ParameterSpace<org.nd4j.linalg.schedule.ISchedule>

getNestedSpaces

public Map<String,ParameterSpace> getNestedSpaces()

Specified by:

getNestedSpaces in interface ParameterSpace<org.nd4j.linalg.schedule.ISchedule>

isLeaf

public boolean isLeaf()

Specified by:

isLeaf in interface ParameterSpace<org.nd4j.linalg.schedule.ISchedule>

setIndices

public void setIndices(int... indices)

Specified by:

setIndices in interface ParameterSpace<org.nd4j.linalg.schedule.ISchedule>