fix

mlr-org · Sep 13, 2024 · f86235b · f86235b
1 parent fff051d
commit f86235b
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 19 deletions.
diff --git a/R/LearnerClassifAuto.R b/R/LearnerClassifAuto.R
@@ -52,7 +52,7 @@ LearnerClassifAuto = R6Class("LearnerClassifAuto",
         small_data_resampling = p_uty(tags = c("train", "super")),
         # cardinality
         max_cardinality = p_int(lower = 1L, default = 100L, tags = c("train", "super")),
-        extra_trees_max_cardinality = p_int(lower = 1L, default = 40L, tags = c("train", "extratrees")),
+        extra_trees_max_cardinality = p_int(lower = 1L, default = 40L, tags = c("train", "extra_trees")),
         # tuner
         resampling = p_uty(tags = c("train", "super")),
         terminator = p_uty(tags = c("train", "super")),

diff --git a/R/LearnerClassifAutoGlmnet.R b/R/LearnerClassifAutoGlmnet.R
@@ -14,24 +14,6 @@ LearnerClassifAutoGlmnet = R6Class("LearnerClassifAutoGlmnet",
     #' Creates a new instance of this [R6][R6::R6Class] class.
     initialize = function(id = "classif.auto_glmnet") {
       super$initialize(id = id, learner_ids = "glmnet")
-
-      # reduce parameter set to the relevant parameters
-      private$.param_set = private$.param_set$subset(
-        c("learner_ids",
-        "learner_timeout",
-        "small_data_size",
-        "small_data_resampling",
-        "max_cardinality",
-        "resampling",
-        "terminator",
-        "measure",
-        "lhs_size",
-        "callbacks",
-        "store_benchmark_result")
-      )
-
-      self$param_set$set_values(learner_ids = "glmnet")
-      self$packages = c("mlr3tuning", "mlr3learners", "mlr3pipelines", "mlr3mbo", "mlr3automl", "glmnet")
     }
   )
 )