set_model_mode

topepo · topepo · commit 96f9adae74c6 · 2021-06-16T10:34:35.000-04:00
diff --git a/R/gen_additive_mod_data.R b/R/gen_additive_mod_data.R
@@ -1,5 +1,7 @@
 
 set_new_model("gen_additive_mod")
+set_model_mode("gen_additive_mod", "classification")
+set_model_mode("gen_additive_mod", "regression")
 
 # ------------------------------------------------------------------------------
 #### REGRESION ----
