The built-in tuning search space of a learner

Returns the search space and default evaluation budget that vimpute(tune = TRUE) uses for one of its built-in learners. The bounds and budgets scale with the task size. Useful as the starting point for a search_space hook in register_vimpute_method() and for inspecting what the tuner explores.

vimpute_search_space(learner_id, task)

Arguments

learner_id

Single string: an mlr3 learner id used by vimpute's built-in methods, e.g. "regr.ranger", "classif.xgboost", "regr.cv_glmnet", "regr.lm_rob", "regr.gam_imp".

task

An mlr3 task (its number of rows scales the bounds and the budget).

Value

A list with elements space (a paradox::ParamSet, or NULL if the learner has no built-in space) and n_evals (the default evaluation budget).

See also

vimpute_tune_control(), register_vimpute_method()

Examples

if (FALSE) { # \dontrun{
task <- mlr3::TaskRegr$new("t", backend = data.frame(y = 1:20, x = 1:20),
                           target = "y")
vimpute_search_space("regr.ranger", task)
} # }