Define a base learner model based on parsnip and tune
Usage
switch_model(
model_type = c("mlp", "xgb", "lgb", "elnet"),
learn_rate = 0.1,
device = "cuda:0"
)Arguments
- model_type
character(1). Model type to be used. Default is "mlp". Available options are "mlp", "xgb", "lgb", "elnet".
- learn_rate
numeric(1). The learning rate for the model. Default is 0.1.
- device
character(1). The device to be used for training. Default is "cuda:0". Make sure that your system is equipped with CUDA-enabled graphical processing units.