Merge pull request #40 from siegfried/tune-sample-size

Use sample_prop instead of sample_size
curso-r · Jul 9, 2021 · 2e1fbdd · 2e1fbdd
2 parents 2e9996a + 08b6274
commit 2e1fbdd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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
   )
 }