momentum: Gradient descent momentum parameter

View source: R/param_momentum.R

momentumR Documentation

Gradient descent momentum parameter

Description

A useful parameter for neural network models using gradient descent

Usage

momentum(range = c(0, 1), trans = NULL)

Arguments

range

A two-element vector holding the defaults for the smallest and largest possible values, respectively. If a transformation is specified, these values should be in the transformed units.

trans

A trans object from the scales package, such as scales::transform_log10() or scales::transform_reciprocal(). If not provided, the default is used which matches the units used in range. If no transformation, NULL.

Examples

momentum()

dials documentation built on Sept. 11, 2024, 8:25 p.m.