Skip to contents

Tuning Parameters for NBEATS Models

Source: R/dials-nbeats_params.R
nbeats_params.Rd

Tuning Parameters for NBEATS Models

Usage

num_stacks(range = c(2, 100), 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::log10_trans() or scales::reciprocal_trans(). If not provided, the default is used which matches the units used in range. If no transformation, NULL.

Details

The main parameters for NBEATS models are:

  • num_stacks: The number of stacks the network should contain.