View source: R/supervised-deep-learning.R
| tl_tune_deep | R Documentation |
Tune a deep learning model
tl_tune_deep(
data,
formula,
is_classification = FALSE,
hidden_layers_options = list(c(32), c(64, 32), c(128, 64, 32)),
learning_rates = c(0.01, 0.001, 1e-04),
batch_sizes = c(16, 32, 64),
epochs = 30,
validation_split = 0.2,
...
)
data |
A data frame containing the training data |
formula |
A formula specifying the model |
is_classification |
Logical indicating if this is a classification problem |
|
List of vectors defining hidden layer configurations to try | |
learning_rates |
Learning rates to try (default: c(0.01, 0.001, 0.0001)) |
batch_sizes |
Batch sizes to try (default: c(16, 32, 64)) |
epochs |
Number of training epochs (default: 30) |
validation_split |
Proportion of data for validation (default: 0.2) |
... |
Additional arguments |
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.