init(learningRate:alpha:epsilon:centered:momentum:gradientScale:regularizationScale:clipsGradientsTo:regularizationFunction:)
Returns a new RMSProp optimizer object.
Declaration
init(learningRate: Float, alpha: Float, epsilon: Float, centered: Bool, momentum: Float, gradientScale: Float, regularizationScale: Float, clipsGradientsTo gradientBounds: ClosedRange<Float>? = nil, regularizationFunction: BNNSOptimizerRegularizationFunction)Parameters
- learningRate:
A value that specifies the learning rate.
- alpha:
A constant that specifies smoothing, in the range
0to1. - epsilon:
A term that the optimizer adds to the denominator.
- centered:
A Boolean value that specifies whether to use the centered variant.
- momentum:
The rate of momentum decay.
- gradientScale:
A value that specifies the gradient scaling factor.
- regularizationScale:
A value that specifies the regularization scaling factor.
- gradientBounds:
The values for the minimum and maximum gradients.
- regularizationFunction:
The variable that specifies the regularization function.