remove bmr_score class

mlr-org · Dec 5, 2024 · a464bb9 · a464bb9
1 parent 003f6e9
commit a464bb9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/R/embedded_ensemble_fselect.R b/R/embedded_ensemble_fselect.R
@@ -88,6 +88,8 @@ embedded_ensemble_fselect = function(
 
   # extract scores on the test sets
   scores = bmr$score(measure)
+  # remove `bmr_score` class
+  class(scores) = c("data.table", "data.frame")
 
   set(scores, j = "features", value = features)
   set(scores, j = "n_features", value = n_features)

diff --git a/R/ensemble_fselect.R b/R/ensemble_fselect.R
@@ -136,6 +136,8 @@ ensemble_fselect = function(
 
   # extract scores on the test sets
   scores = bmr$score(measure)
+  # remove `bmr_score` class
+  class(scores) = c("data.table", "data.frame")
 
   set(scores, j = "features", value = features)
   set(scores, j = "n_features", value = n_features)