org.nd4j.linalg.schedule

Class PolySchedule

java.lang.Object

org.nd4j.linalg.schedule.PolySchedule

All Implemented Interfaces:

Serializable, Cloneable, ISchedule

public class PolySchedule
extends Object
implements ISchedule

Polynomial decay schedule, with 3 parameters: initial value, maxIter, power.
Note that the value will be 0 after maxIter, otherwise is given by: value(i) = initialValue * (1 + i/maxIter)^(-power) where i is the iteration or epoch (depending on the setting)

Author:

Alex Black

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
PolySchedule(ScheduleType scheduleType, double initialValue, double power, int maxIter)

Method Summary

Modifier and Type	Method and Description
PolySchedule	clone()
double	valueAt(int iteration, int epoch)

Methods inherited from class java.lang.Object

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

Constructor Detail

PolySchedule

public PolySchedule(ScheduleType scheduleType,
                    double initialValue,
                    double power,
                    int maxIter)

Method Detail

valueAt

public double valueAt(int iteration,
                      int epoch)

Specified by:

valueAt in interface ISchedule

Parameters:

iteration - Current iteration number. Starts at 0

epoch - Current epoch number. Starts at 0

Returns:

Value at the current iteration/epoch for this schedule

clone

public PolySchedule clone()

Specified by:

clone in interface ISchedule

Overrides:

clone in class Object