feat: allow callbacks

mlr-org · Apr 10, 2024 · 2e05bd3 · 2e05bd3
1 parent 2b724ae
commit 2e05bd3
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/R/ensemble_fselect.R b/R/ensemble_fselect.R
@@ -30,14 +30,10 @@
 #'     terminator = trm("evals", n_evals = 10)
 #'   )
 #' }
-ensemble_fselect = function(fselector, task, learners, outer_resampling, inner_resampling, measure, terminator) {
+ensemble_fselect = function(fselector, task, learners, outer_resampling, inner_resampling, measure, terminator, callbacks = list()) {
   assert_task(task)
   assert_learners(as_learners(learners), task = task)
   assert_resampling(outer_resampling)
-  assert_resampling(inner_resampling)
-  assert_measure(measure)
-  assert_fselector(fselector)
-  assert_terminator(terminator)
 
   # create fselector for each learner
   afss = map(learners, function(learner) {
@@ -47,7 +43,8 @@ ensemble_fselect = function(fselector, task, learners, outer_resampling, inner_r
       resampling = inner_resampling,
       measure = measure,
       terminator = terminator,
-      store_models = TRUE
+      store_models = TRUE,
+      callbacks = callbacks
     )
   })