add model args for naive_Bayes(engine = "linfa")

simonpcouch · Jul 29, 2024 · 4e620bc · 4e620bc
1 parent 4db6ab2
commit 4e620bc
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/R/mod-naive_Bayes.R b/R/mod-naive_Bayes.R
@@ -81,6 +81,15 @@ make_naive_Bayes_linfa <- function() {
     )
   )
 
+  parsnip::set_model_arg(
+    model = "naive_Bayes",
+    eng = "linfa",
+    parsnip = "smoothness",
+    original = "smoothness",
+    func = list(pkg = "dials", fun = "smoothness"),
+    has_submodel = FALSE
+  )
+
   parsnip::set_encoding(
     model = "naive_Bayes",
     mode = "classification",