View source: R/supervised-deep-learning.R
| tl_fit_deep | R Documentation |
Fit a deep learning model
tl_fit_deep(
data,
formula,
is_classification = FALSE,
hidden_layers = c(32, 16),
activation = "relu",
dropout = 0.2,
epochs = 30,
batch_size = 32,
validation_split = 0.2,
verbose = 0,
...
)
data |
A data frame containing the training data |
formula |
A formula specifying the model |
is_classification |
Logical indicating if this is a classification problem |
|
Vector of units in each hidden layer (default: c(32, 16)) | |
activation |
Activation function for hidden layers (default: "relu") |
dropout |
Dropout rate for regularization (default: 0.2) |
epochs |
Number of training epochs (default: 30) |
batch_size |
Batch size for training (default: 32) |
validation_split |
Proportion of data for validation (default: 0.2) |
verbose |
Verbosity mode (0 = silent, 1 = progress bar, 2 = one line per epoch) (default: 0) |
... |
Additional arguments |
A fitted deep learning model
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.