RMSProp

class RMSProp(learningRate: Float, decay: Float, momentum: Float, epsilon: Float, centered: Boolean, clipGradient: ClipGradientAction) : Optimizer

RMSProp optimizer.

Constructors

RMSProp

fun RMSProp(learningRate: Float = 0.001f, decay: Float = 0.9f, momentum: Float = 0.0f, epsilon: Float = 1e-10f, centered: Boolean = false, clipGradient: ClipGradientAction = NoClipGradient())

clipGradient

Strategy of gradient clipping as sub-class of ClipGradientAction.

optimizerName

open override val optimizerName: String

Returns optimizer name.