From 08b6274f3423aafd59a532b42e32f06a6f2808df Mon Sep 17 00:00:00 2001 From: Zhi-Qiang Lei Date: Tue, 1 Jun 2021 09:36:24 +0800 Subject: [PATCH] Use sample_prop instead of sample_size Only values between 0 to 1 are allowed. --- R/lightgbm.R | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/R/lightgbm.R b/R/lightgbm.R index 227e2d1..61ea6f5 100644 --- a/R/lightgbm.R +++ b/R/lightgbm.R @@ -168,7 +168,7 @@ add_boost_tree_lightgbm <- function() { eng = "lightgbm", parsnip = "sample_size", original = "bagging_fraction", - func = list(pkg = "dials", fun = "sample_size"), + func = list(pkg = "dials", fun = "sample_prop"), has_submodel = FALSE ) }