From 9720e0b0d029f41f8daf8e1dc75c1df1cef7ba19 Mon Sep 17 00:00:00 2001
From: Daniel Falbel <dfalbel@gmail.com>
Date: Fri, 9 Jul 2021 18:52:17 -0300
Subject: [PATCH] port fix from #46 for tree

---
 R/tree.R | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/R/tree.R b/R/tree.R
index 79d2eec..32d1b08 100644
--- 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",