compute time elapsed for both fit_xy and fit and print

Author: patr1ckm

R/fit.R

@@ -368,6 +368,8 @@ check_xy_interface <- function(x, y, cl, model) {
 #' @export
 print.model_fit <- function(x, ...) {
   cat("parsnip model object\n\n")
+  cat("Fit in: ", paste(format(unclass(x$elapsed), digits = 2),
+                        attr(x$elapsed, "units")))
 
   if (inherits(x$fit, "try-error")) {
     cat("Model fit failed with error:\n", x$fit, "\n")

R/fit_helpers.R

@@ -52,14 +52,19 @@ form_form <-
       spec = object
     )
 
+    start <- Sys.time()
+
     res$fit <- eval_mod(
       fit_call,
       capture = control$verbosity == 0,
       catch = control$catch,
       env = env,
       ...
     )
+
+    end <- Sys.time()
     res$preproc <- list(y_var = all.vars(env$formula[[2]]))
+    res$elapsed <- end - start
     res
   }
 
@@ -107,19 +112,25 @@ xy_xy <- function(object, env, control, target = "none", ...) {
 
   res <- list(lvl = levels(env$y), spec = object)
 
+  start <- Sys.time()
+
   res$fit <- eval_mod(
     fit_call,
     capture = control$verbosity == 0,
     catch = control$catch,
     env = env,
     ...
   )
+
+  end <- Sys.time()
+
   if (is.vector(env$y)) {
     y_name <- character(0)
   } else {
     y_name <- colnames(env$y)
   }
   res$preproc <- list(y_var = y_name)
+  res$elapsed <- end - start
   res
 }
 

tests/testthat/test_fit_interfaces.R

@@ -79,3 +79,20 @@ test_that('unknown modes', {
   )
 })
 
+test_that("elapsed time parsnip mods", {
+  lm1 <-
+    linear_reg() %>%
+    set_engine("lm") %>%
+    fit_xy(x = mtcars[, 2:4], y = mtcars$mpg)
+
+  lm2 <-
+    linear_reg() %>%
+    set_engine("lm") %>%
+    fit(mpg ~ ., data = mtcars)
+
+  expect_output(print(lm1), "Fit in:")
+  expect_output(print(lm2), "Fit in:")
+  expect_true(!is.null(lm1$elapsed))
+  expect_true(!is.null(lm2$elapsed))
+})
+