dials (version 0.0.9)

dropout: Neural network parameters

Description

These functions generate parameters that are useful for neural network models.

Usage

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

epochs(range = c(10L, 1000L), trans = NULL)

hidden_units(range = c(1L, 10L), trans = NULL)

batch_size(range = c(unknown(), unknown()), trans = log2_trans())

Arguments

range

A two-element vector holding the defaults for the smallest and largest possible values, respectively.

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

  • dropout(): The parameter dropout rate. (See parsnip:::mlp()).

  • epochs(): The number of iterations of training. (See parsnip:::mlp()).

  • hidden_units(): The number of hidden units in a network layer. (See parsnip:::mlp()).

  • batch_size(): The mini-batch size for neural networks.

Examples

Run this code
# NOT RUN {
dropout()
# }

Run the code above in your browser using DataCamp Workspace