expand_modelgrid(TunedModel(GBMModel, grid = 5))
expand_modelgrid(TunedModel(GLMNetModel, grid = c(alpha = 5, lambda = 10)),
sale_amount ~ ., data = ICHomes)
gbm_grid <- ParameterGrid(
n.trees = dials::trees(),
interaction.depth = dials::tree_depth(),
size = 5
)
expand_modelgrid(TunedModel(GBMModel, grid = gbm_grid))
rf_grid <- ParameterGrid(
mtry = dials::mtry(),
nodesize = dials::max_nodes(),
size = c(3, 5)
)
expand_modelgrid(TunedModel(RandomForestModel, grid = rf_grid),
sale_amount ~ ., data = ICHomes)
Run the code above in your browser using DataLab