port fix from #46 for tree

curso-r · Jul 9, 2021 · 9720e0b · 9720e0b
1 parent abaac2e
commit 9720e0b
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/R/tree.R b/R/tree.R
@@ -12,6 +12,18 @@ add_decision_tree_tree <- function() {
 
   parsnip::set_dependency("decision_tree", eng = "tree", pkg = "tree")
 
+  parsnip::set_encoding(
+    model = "decision_tree",
+    eng = "tree",
+    mode = "regression",
+    options = list(
+      predictor_indicators = "none",
+      compute_intercept = FALSE,
+      remove_intercept = FALSE,
+      allow_sparse_x = FALSE
+    )
+  )
+
   parsnip::set_fit(
     model = "decision_tree",
     eng = "tree",