RMSpropOptimizer

An optimizer that implements the RMSprop optimization algorithm.

Constructors

RMSpropOptimizer

fun RMSpropOptimizer(alpha: Float = 0.005f, beta: Float = 0.9f)

tensorTrainingStep

open override fun tensorTrainingStep(tensor: DTensor, gradient: DTensor): DTensor

Train an element of the model. For the model itself, use train.

train

fun train(component: T, tangent: Trainable.Tangent): T

Train a trainable using the given tangent.

alpha

val alpha: Float = 0.005f

beta

val beta: Float = 0.9f