add tidy() method for model_fit objects

Author: malcolmbarrett

DESCRIPTION

@@ -29,7 +29,7 @@ Imports:
     tidyr,
     globals
 Roxygen: list(markdown = TRUE)
-RoxygenNote: 6.1.0.9000
+RoxygenNote: 6.1.1
 Suggests: 
     testthat,
     knitr,
@@ -39,4 +39,3 @@ Suggests:
     xgboost,
     covr,
     sparklyr
-

NAMESPACE

@@ -119,6 +119,7 @@ export(show_call)
 export(surv_reg)
 export(svm_poly)
 export(svm_rbf)
+export(tidy.model_fit)
 export(translate)
 export(varying)
 export(varying_args)

R/tidy.R

@@ -0,0 +1,22 @@
+#' Turn a parsnip model object into a tidy tibble
+#'
+#' This method tidies the model in a parsnip model object, if it exists.
+#'
+#' @inheritParams generics::tidy
+#'
+#' @return a tibble
+#' @export
+#'
+#' @examples
+#' logistic_reg() %>%
+#'   set_engine("glm") %>%
+#'   fit(Class ~ funded_amnt + int_rate, data = lending_club) %>%
+#'   # tidying model object and passing arguments to broom:::tidy.glm
+#'   tidy(conf.int = TRUE, exponentiate = TRUE)
+tidy.model_fit <- function(x, ...) {
+  tryCatch(generics::tidy(x$fit, ...),
+    error = function(error) {
+      cat("Error:", error$message)
+    }
+  )
+}