Contents

init(descriptor:momentumScale:alpha:epsilon:isCentered:)

Creates an RMSProp optimizer with the descriptor, momentum scale, smoothing, epsilon, and option to compute the centered RMSProp that you specify.

Declaration

convenience init(descriptor optimizerDescriptor: MLCOptimizerDescriptor, momentumScale: Float, alpha: Float, epsilon: Float, isCentered: Bool)

Parameters

  • optimizerDescriptor:

    An object you use to configure the optimizer.

  • momentumScale:

    The momentum scale.

  • alpha:

    The smoothing constant value.

  • epsilon:

    The epsilon value you use to improve numerical stability.

  • isCentered:

    A Boolean that indicates whether you compute the centered RMSProp.

See Also

Creating an RMSProp Optimizer