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

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