View source: R/supervised-neural-networks.R
| tl_tune_nn | R Documentation |
Tune a neural network model
tl_tune_nn(
data,
formula,
is_classification = FALSE,
sizes = c(1, 2, 5, 10),
decays = c(0, 0.001, 0.01, 0.1),
folds = 5,
...
)
data |
A data frame containing the training data |
formula |
A formula specifying the model |
is_classification |
Logical indicating if this is a classification problem |
sizes |
Vector of hidden layer sizes to try |
decays |
Vector of weight decay parameters to try |
folds |
Number of cross-validation folds (default: 5) |
... |
Additional arguments to pass to nnet() |
A list with the best model and tuning results
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.