catboost parameter updates (#119)

Co-authored-by: Emil Hvitfeldt <[email protected]>

Author: topepo

.Rbuildignore

@@ -12,4 +12,5 @@ inst/figs/
 ^README\.Rmd$
 ^[\.]?air\.toml$
 ^\.vscode$
-^revdep$
+^revdep$
+^train_dir$

.gitignore

@@ -11,3 +11,4 @@ revdep/library.noindex
 revdep/data.sqlite
 .httr-oauth
 revdep/cloud.noindex/*
+train_dir/

DESCRIPTION

@@ -26,7 +26,7 @@ License: MIT + file LICENSE
 URL: https://bonsai.tidymodels.org/, https://github.com/tidymodels/bonsai
 BugReports: https://github.com/tidymodels/bonsai/issues
 Depends:
-    parsnip (>= 1.0.1),
+    parsnip (>= 1.3.3.9000),
     R (>= 4.1)
 Imports: 
     cli,
@@ -52,9 +52,11 @@ Suggests:
     tune
 VignetteBuilder: 
     knitr
+Remotes:
+    tidymodels/parsnip
 Config/Needs/website: tidyverse/tidytemplate
 Config/testthat/edition: 3
 Config/usethis/last-upkeep: 2025-04-25
 Encoding: UTF-8
 Roxygen: list(markdown = TRUE)
-RoxygenNote: 7.3.2
+RoxygenNote: 7.3.3

NEWS.md

@@ -1,5 +1,9 @@
 # bonsai (development version)
 
+* catboost tuning parameters were updates: (#119)
+  - Fixed a bug in `mtry`; it used the parsnip name `mtry_prop`.
+  - Enabled model arguments for `min_n`, `sample_size`, and `stop_iter`. `min_n` is only enabled for GPU computations. 
+  
 # bonsai 0.4.0
 
 ## New Engines

R/catboost_data.R

@@ -184,11 +184,38 @@ make_boost_tree_catboost <- function() {
   parsnip::set_model_arg(
     model = "boost_tree",
     eng = "catboost",
-    parsnip = "mtry_prop",
+    parsnip = "mtry",
     original = "rsm",
     func = list(pkg = "dials", fun = "mtry_prop"),
     has_submodel = FALSE
   )
+
+  parsnip::set_model_arg(
+    model = "boost_tree",
+    eng = "catboost",
+    parsnip = "min_n",
+    original = "min_data_in_leaf",
+    func = list(pkg = "dials", fun = "min_n"),
+    has_submodel = FALSE
+  )
+
+  parsnip::set_model_arg(
+    model = "boost_tree",
+    eng = "catboost",
+    parsnip = "sample_size",
+    original = "subsample",
+    func = list(pkg = "dials", fun = "sample_prop"),
+    has_submodel = FALSE
+  )
+
+  parsnip::set_model_arg(
+    model = "boost_tree",
+    eng = "catboost",
+    parsnip = "stop_iter",
+    original = "early_stopping_rounds",
+    func = list(pkg = "dials", fun = "stop_iter"),
+    has_submodel = FALSE
+  )
 }
 
 # nocov end