public class NesterovsSpace extends BaseUpdaterSpace
| Modifier and Type | Field and Description |
|---|---|
protected ParameterSpace<Double> |
learningRate |
protected ParameterSpace<ISchedule> |
learningRateSchedule |
protected ParameterSpace<Double> |
momentum |
protected ParameterSpace<ISchedule> |
momentumSchedule |
| Constructor and Description |
|---|
NesterovsSpace(ParameterSpace<Double> learningRate) |
NesterovsSpace(ParameterSpace<Double> learningRate,
ParameterSpace<Double> momentum) |
NesterovsSpace(ParameterSpace<Double> learningRate,
ParameterSpace<ISchedule> learningRateSchedule,
ParameterSpace<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<Double> lr) |
static NesterovsSpace |
withLR(ParameterSpace<Double> lr,
double momentum) |
static NesterovsSpace |
withLR(ParameterSpace<Double> lr,
ParameterSpace<Double> momentum) |
static NesterovsSpace |
withLRSchedule(ParameterSpace<ISchedule> lrSchedule) |
static NesterovsSpace |
withLRSchedule(ParameterSpace<ISchedule> lrSchedule,
double momentum) |
static NesterovsSpace |
withLRSchedule(ParameterSpace<ISchedule> lrSchedule,
ParameterSpace<Double> momentum) |
collectLeaves, isLeaf, numParameters, setIndicesgetNestedSpacesprotected ParameterSpace<Double> learningRate
protected ParameterSpace<ISchedule> learningRateSchedule
protected ParameterSpace<Double> momentum
protected ParameterSpace<ISchedule> momentumSchedule
public NesterovsSpace(ParameterSpace<Double> learningRate)
public NesterovsSpace(ParameterSpace<Double> learningRate, ParameterSpace<Double> momentum)
public NesterovsSpace(ParameterSpace<Double> learningRate, ParameterSpace<ISchedule> learningRateSchedule, ParameterSpace<Double> momentum, ParameterSpace<ISchedule> momentumSchedule)
public static NesterovsSpace withLR(ParameterSpace<Double> lr)
public static NesterovsSpace withLR(ParameterSpace<Double> lr, double momentum)
public static NesterovsSpace withLR(ParameterSpace<Double> lr, ParameterSpace<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<Double> momentum)
public IUpdater getValue(double[] parameterValues)
ParameterSpaceparameterValues - A set of values, each in the range [0,1], of length ParameterSpace.numParameters()Copyright © 2020. All rights reserved.