diff --git a/R/embedded_ensemble_fselect.R b/R/embedded_ensemble_fselect.R index ce62b87d..94ca4ac6 100644 --- a/R/embedded_ensemble_fselect.R +++ b/R/embedded_ensemble_fselect.R @@ -85,6 +85,9 @@ embedded_ensemble_fselect = function( ) }) + # TOCHECK: can we have `task` as the original task here, + # `resampling` the subsamples (train/test partitions) so that + # bmr$score(measure) later just works out of the box? design = grid[, c("learner", "task", "resampling"), with = FALSE] bmr = benchmark(design, store_models = TRUE)