crumble_control {crumble} | R Documentation |
Crumble control parameters
Description
Crumble control parameters
Usage
crumble_control(
crossfit_folds = 10L,
mlr3superlearner_folds = 10L,
zprime_folds = 1L,
epochs = 100L,
learning_rate = 0.01,
batch_size = 64,
device = c("cpu", "cuda", "mps")
)
Arguments
crossfit_folds |
[ |
mlr3superlearner_folds |
[ |
zprime_folds |
[ |
epochs |
[ |
learning_rate |
[ |
batch_size |
[ |
device |
[ |
Value
A list of control parameters
Examples
if (torch::torch_is_installed()) crumble_control(crossfit_folds = 5)
[Package crumble version 0.1.2 Index]