Merge pull request #98 from mlr-org/RaphaelS1-patch-1

Fix bart base

Author: RaphaelS1

DESCRIPTION

@@ -1,6 +1,6 @@
 Package: mlr3extralearners
 Title: Extra Learners For mlr3
-Version: 0.5.0
+Version: 0.5.1
 Authors@R:
     c(person(given = "Raphael",
              family = "Sonabend",

NEWS.md

@@ -1,3 +1,7 @@
+# mlr3extralearners 0.5.1
+
+* Fixes bug in `base` parameter of {bart} learners
+
 # mlr3extralearners 0.5.0
 
 * Deprecated liblinear learners now removed

R/learner_dbarts_classif_bart.R

@@ -21,7 +21,7 @@ LearnerClassifBart = R6Class("LearnerClassifBart",
         ntree = p_int(default = 200L, lower = 1L, tags = "train"),
         k = p_dbl(default = 2.0, lower = 0, tags = "train"),
         power = p_dbl(default = 2.0, lower = 0, tags = "train"),
-        base = p_dbl(default = 0.95, lower = 0, tags = "train"),
+        base = p_dbl(default = 0.95, lower = 0, upper = 1, tags = "train"),
         binaryOffset = p_dbl(default = 0.0, tags = "train"),
         ndpost = p_int(default = 1000L, lower = 1L, tags = "train"),
         nskip = p_int(default = 100L, lower = 0L, tags = "train"),

R/learner_dbarts_regr_bart.R

@@ -30,7 +30,7 @@ LearnerRegrBart <- R6Class("LearnerRegrBart",
         sigquant = p_dbl(default = 0.90, lower = 0, upper = 1, tags = "train"),
         k = p_dbl(default = 2.0, lower = 0, tags = "train"),
         power = p_dbl(default = 2.0, lower = 0, tags = "train"),
-        base = p_dbl(default = 0.95, lower = 0, tags = "train"),
+        base = p_dbl(default = 0.95, lower = 0, upper = 1, tags = "train"),
         ndpost = p_int(default = 1000L, lower = 1L, tags = "train"),
         nskip = p_int(default = 100L, lower = 0L, tags = "train"),
         printevery = p_int(default = 100L, lower = 0L, tags = "train"),