Merge pull request #833 from tidymodels/move-fit_xy.decision_tree

add `fit_xy.decision_tree()`

Author: hfrick

DESCRIPTION

@@ -1,6 +1,6 @@
 Package: parsnip
 Title: A Common API to Modeling and Analysis Functions
-Version: 1.0.2.9001
+Version: 1.0.2.9002
 Authors@R: c(
     person("Max", "Kuhn", , "[email protected]", role = c("aut", "cre")),
     person("Davis", "Vaughan", , "[email protected]", role = "aut"),

NAMESPACE

@@ -6,6 +6,7 @@ S3method(extract_parameter_dials,model_spec)
 S3method(extract_parameter_set_dials,model_spec)
 S3method(extract_spec_parsnip,model_fit)
 S3method(fit,model_spec)
+S3method(fit_xy,decision_tree)
 S3method(fit_xy,gen_additive_mod)
 S3method(fit_xy,model_spec)
 S3method(glance,model_fit)

R/decision_tree.R

@@ -134,6 +134,27 @@ check_args.decision_tree <- function(object) {
 
 # ------------------------------------------------------------------------------
 
+#' @export
+fit_xy.decision_tree <- function(object,
+                                 x,
+                                 y,
+                                 case_weights = NULL,
+                                 control = parsnip::control_parsnip(),
+                                 ...) {
+
+  if (object$mode == "censored regression" && object$engine == "rpart") {
+    # prodlim::EventHistory.frame() expects a call to `Surv()` (or `Hist()`) on
+    # the left-hand side of the formula
+    rlang::abort("For the `'rpart'` engine, please use the formula interface via `fit()`.")
+  }
+
+  # call parsnip::fit_xy.model_spec()
+  res <- NextMethod()
+  res
+}
+
+# ------------------------------------------------------------------------------
+
 #' Decision trees via rpart
 #'
 #' `rpart_train` is a wrapper for `rpart()` tree-based models