Skip to content

Commit 2ac9743

Browse files
committed
catboost parameter updates
1 parent 4adea7a commit 2ac9743

File tree

4 files changed

+32
-4
lines changed

4 files changed

+32
-4
lines changed

.Rbuildignore

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,4 +12,5 @@ inst/figs/
1212
^README\.Rmd$
1313
^[\.]?air\.toml$
1414
^\.vscode$
15-
^revdep$
15+
^revdep$
16+
^train_dir$

DESCRIPTION

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,4 +57,4 @@ Config/testthat/edition: 3
5757
Config/usethis/last-upkeep: 2025-04-25
5858
Encoding: UTF-8
5959
Roxygen: list(markdown = TRUE)
60-
RoxygenNote: 7.3.2
60+
RoxygenNote: 7.3.3

R/catboost_data.R

Lines changed: 28 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -184,11 +184,38 @@ make_boost_tree_catboost <- function() {
184184
parsnip::set_model_arg(
185185
model = "boost_tree",
186186
eng = "catboost",
187-
parsnip = "mtry_prop",
187+
parsnip = "mtry",
188188
original = "rsm",
189189
func = list(pkg = "dials", fun = "mtry_prop"),
190190
has_submodel = FALSE
191191
)
192+
193+
parsnip::set_model_arg(
194+
model = "boost_tree",
195+
eng = "catboost",
196+
parsnip = "min_n",
197+
original = "min_data_in_leaf",
198+
func = list(pkg = "dials", fun = "min_n"),
199+
has_submodel = FALSE
200+
)
201+
202+
parsnip::set_model_arg(
203+
model = "boost_tree",
204+
eng = "catboost",
205+
parsnip = "sample_size",
206+
original = "subsample",
207+
func = list(pkg = "dials", fun = "sample_prop"),
208+
has_submodel = FALSE
209+
)
210+
211+
parsnip::set_model_arg(
212+
model = "boost_tree",
213+
eng = "catboost",
214+
parsnip = "stop_iter",
215+
original = "early_stopping_rounds",
216+
func = list(pkg = "dials", fun = "stop_iter"),
217+
has_submodel = FALSE
218+
)
192219
}
193220

194221
# nocov end

man/bonsai-package.Rd

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)