Merge pull request #247 from tidymodels/devel

use modeldata package and data treatment in predict

Author: topepo

DESCRIPTION

@@ -49,4 +49,5 @@ Suggests:
     ranger,
     rpart,
     MASS,
-    nlme
+    nlme,
+    modeldata

R/data.R

@@ -58,6 +58,7 @@
 #'
 #' \preformatted{
 #'
+#' library(modeldata)
 #' data("lending_club")
 #'
 #' rand_forest(mode = "classification", mtry = .cols() - 2)

R/descriptors.R

@@ -48,6 +48,7 @@
 #' # methods for specifying the model can be used
 #'
 #' library(dplyr)
+#' library(modeldata)
 #' data("lending_club")
 #'
 #' lr_mod <- logistic_reg()

R/fit.R

@@ -229,7 +229,12 @@ prepare_data <- function(object, new_data) {
       new_data <- convert_form_to_xy_new(object$preproc, new_data)$x
     }
   }
-
-  new_data
+  switch(
+    fit_interface,
+    none = new_data,
+    data.frame = as.data.frame(new_data),
+    matrix = as.matrix(new_data),
+    new_data
+  )
 }