Skip to content

Commit 1d2ed35

Browse files
committed
sample_size to sample_prop avoid gte 1 error
1 parent 57b5f56 commit 1d2ed35

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

R/catboost.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -189,7 +189,7 @@ add_boost_tree_catboost <- function() {
189189
parsnip::set_model_arg(
190190
model = "boost_tree",
191191
eng = "catboost",
192-
parsnip = "sample_size",
192+
parsnip = "sample_prop",
193193
original = "subsample",
194194
func = list(pkg = "dials", fun = "sample_size"),
195195
has_submodel = FALSE

R/lightgbm.R

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -166,7 +166,7 @@ add_boost_tree_lightgbm <- function() {
166166
parsnip::set_model_arg(
167167
model = "boost_tree",
168168
eng = "lightgbm",
169-
parsnip = "sample_size",
169+
parsnip = "sample_prop",
170170
original = "bagging_fraction",
171171
func = list(pkg = "dials", fun = "sample_size"),
172172
has_submodel = FALSE

0 commit comments

Comments
 (0)