diff --git a/DESCRIPTION b/DESCRIPTION index 0b96839e3..b6fd20253 100644 --- a/DESCRIPTION +++ b/DESCRIPTION @@ -1,6 +1,6 @@ Package: mlr3extralearners Title: Extra Learners For mlr3 -Version: 0.5.9 +Version: 0.5.10 Authors@R: c(person(given = "Raphael", family = "Sonabend", diff --git a/NEWS.md b/NEWS.md index 4f51908d8..26b156e0d 100644 --- a/NEWS.md +++ b/NEWS.md @@ -1,3 +1,7 @@ +# mlr3extralearners 0.5.10 + +* Added params `ignored_features` and `one_hot_max_size` to `classif.catboost` + # mlr3extralearners 0.5.9 * Fixed bug that didn't allow C parameter to be set for nu-regression diff --git a/R/learner_catboost_classif_catboost.R b/R/learner_catboost_classif_catboost.R index 37e559ae6..6abf187b1 100644 --- a/R/learner_catboost_classif_catboost.R +++ b/R/learner_catboost_classif_catboost.R @@ -73,8 +73,8 @@ LearnerClassifCatboost = R6Class("LearnerClassifCatboost", grow_policy = p_fct(levels = c("SymmetricTree", "Depthwise", "Lossguide"), default = "SymmetricTree", tags = "train"), min_data_in_leaf = p_int(lower = 1L, upper = Inf, default = 1L, tags = "train"), max_leaves = p_int(lower = 1L, upper = Inf, default = 31L, tags = "train"), - # ignored_features missing - # one_hot_max_size missing + ignored_features = p_uty(default = NULL, tags = "train"), + one_hot_max_size = p_uty(default = FALSE, tags = "train"), has_time = p_lgl(default = FALSE, tags = "train"), rsm = p_dbl(lower = 0.001, upper = 1, default = 1, tags = "train"), nan_mode = p_fct(levels = c("Min", "Max"), default = "Min", tags = "train"), # do not allow "Forbidden"