feat: add min.bucket parameter to ranger learner (#266)

* feat: add min.bucket parameter to ranger learner * chore: update news
mlr-org · Apr 17, 2023 · 165b90b · 165b90b
1 parent bedb629
commit 165b90b
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/NEWS.md b/NEWS.md
@@ -1,6 +1,7 @@
 # mlr3learners 0.5.6-9000
 
 * Added formula argument to `nnet` learner and support feature type `"integer"`
+* Added `min.bucket` parameter to `classif.ranger` and `regr.ranger`.
 
 # mlr3learners 0.5.6
 

diff --git a/R/LearnerClassifRanger.R b/R/LearnerClassifRanger.R
@@ -44,6 +44,7 @@ LearnerClassifRanger = R6Class("LearnerClassifRanger",
         importance                   = p_fct(c("none", "impurity", "impurity_corrected", "permutation"), tags = "train"),
         keep.inbag                   = p_lgl(default = FALSE, tags = "train"),
         max.depth                    = p_int(default = NULL, lower = 0L, special_vals = list(NULL), tags = "train"),
+        min.bucket                   = p_int(1L, default = 1L, tags = "train"),
         min.node.size                = p_int(1L, default = NULL, special_vals = list(NULL), tags = "train"),
         min.prop                     = p_dbl(default = 0.1, tags = "train"),
         minprop                      = p_dbl(default = 0.1, tags = "train"),
@@ -131,8 +132,10 @@ LearnerClassifRanger = R6Class("LearnerClassifRanger",
       newdata = ordered_features(task, self)
 
       prediction = invoke(predict,
-        self$model, data = newdata,
-        predict.type = "response", .args = pv)
+        self$model,
+        data = newdata,
+        predict.type = "response", .args = pv
+      )
 
       if (self$predict_type == "response") {
         list(response = prediction$predictions)

diff --git a/R/LearnerRegrRanger.R b/R/LearnerRegrRanger.R
@@ -33,6 +33,7 @@ LearnerRegrRanger = R6Class("LearnerRegrRanger",
         importance                   = p_fct(c("none", "impurity", "impurity_corrected", "permutation"), tags = "train"),
         keep.inbag                   = p_lgl(default = FALSE, tags = "train"),
         max.depth                    = p_int(default = NULL, lower = 0L, special_vals = list(NULL), tags = "train"),
+        min.bucket                   = p_int(1L, default = 1L, tags = "train"),
         min.node.size                = p_int(1L, default = 5L, special_vals = list(NULL), tags = "train"),
         min.prop                     = p_dbl(default = 0.1, tags = "train"),
         minprop                      = p_dbl(default = 0.1, tags = "train"),