public class NesterovsSpace extends BaseUpdaterSpace
Modifier and Type | Field and Description |
---|---|
protected ParameterSpace<java.lang.Double> |
learningRate |
protected ParameterSpace<ISchedule> |
learningRateSchedule |
protected ParameterSpace<java.lang.Double> |
momentum |
protected ParameterSpace<ISchedule> |
momentumSchedule |
Constructor and Description |
---|
NesterovsSpace(ParameterSpace<java.lang.Double> learningRate) |
NesterovsSpace(ParameterSpace<java.lang.Double> learningRate,
ParameterSpace<java.lang.Double> momentum) |
NesterovsSpace(ParameterSpace<java.lang.Double> learningRate,
ParameterSpace<ISchedule> learningRateSchedule,
ParameterSpace<java.lang.Double> momentum,
ParameterSpace<ISchedule> momentumSchedule) |
Modifier and Type | Method and Description |
---|---|
IUpdater |
getValue(double[] parameterValues)
Generate a candidate given a set of values.
|
static NesterovsSpace |
withLR(ParameterSpace<java.lang.Double> lr) |
static NesterovsSpace |
withLR(ParameterSpace<java.lang.Double> lr,
double momentum) |
static NesterovsSpace |
withLR(ParameterSpace<java.lang.Double> lr,
ParameterSpace<java.lang.Double> momentum) |
static NesterovsSpace |
withLRSchedule(ParameterSpace<ISchedule> lrSchedule) |
static NesterovsSpace |
withLRSchedule(ParameterSpace<ISchedule> lrSchedule,
double momentum) |
static NesterovsSpace |
withLRSchedule(ParameterSpace<ISchedule> lrSchedule,
ParameterSpace<java.lang.Double> momentum) |
collectLeaves, isLeaf, numParameters, setIndices
getNestedSpaces
protected ParameterSpace<java.lang.Double> learningRate
protected ParameterSpace<ISchedule> learningRateSchedule
protected ParameterSpace<java.lang.Double> momentum
protected ParameterSpace<ISchedule> momentumSchedule
public NesterovsSpace(ParameterSpace<java.lang.Double> learningRate)
public NesterovsSpace(ParameterSpace<java.lang.Double> learningRate, ParameterSpace<java.lang.Double> momentum)
public NesterovsSpace(ParameterSpace<java.lang.Double> learningRate, ParameterSpace<ISchedule> learningRateSchedule, ParameterSpace<java.lang.Double> momentum, ParameterSpace<ISchedule> momentumSchedule)
public static NesterovsSpace withLR(ParameterSpace<java.lang.Double> lr)
public static NesterovsSpace withLR(ParameterSpace<java.lang.Double> lr, double momentum)
public static NesterovsSpace withLR(ParameterSpace<java.lang.Double> lr, ParameterSpace<java.lang.Double> momentum)
public static NesterovsSpace withLRSchedule(ParameterSpace<ISchedule> lrSchedule)
public static NesterovsSpace withLRSchedule(ParameterSpace<ISchedule> lrSchedule, double momentum)
public static NesterovsSpace withLRSchedule(ParameterSpace<ISchedule> lrSchedule, ParameterSpace<java.lang.Double> momentum)
public IUpdater getValue(double[] parameterValues)
ParameterSpace
parameterValues
- A set of values, each in the range [0,1], of length ParameterSpace.numParameters()