add extract_fit_time

[EmilHvitfeldt]

This is part of a series of PRs to add a new generic extract_fit_time() to allow users to investigate how long different tidymodels objects takes to fit.

TODO:

• make sure names are good.
  • right now id clashed with id returned from tune_grid()
• add tests
• Make sure this change is backwards compatible
• remove hardhat from Remotes

Related PRs
tidymodels/hardhat#218
tidymodels/parsnip#853
tidymodels/workflows#191

library(tidymodels)

data(ames, package = "modeldata")

rec_ames <- recipe(Sale_Price ~ ., data = ames) |>
  step_dummy(all_nominal_predictors()) |>
  step_nzv(all_predictors()) |>
  step_normalize(all_predictors()) |>
  prep()

rec_ames |>
  extract_fit_time()
#> # A tibble: 6 × 3
#>   id              part     time
#>   <chr>           <chr>   <dbl>
#> 1 dummy_xmsLC     prep  0.0230 
#> 2 dummy_xmsLC     bake  0.0290 
#> 3 nzv_sUtzk       prep  0.296  
#> 4 nzv_sUtzk       bake  0      
#> 5 normalize_t06mp prep  0.00700
#> 6 normalize_t06mp bake  0.00300

rec_ames |>
  extract_fit_time() |>
  count(id, wt = time, name = "time")
#> # A tibble: 3 × 2
#>   id                time
#>   <chr>            <dbl>
#> 1 dummy_xmsLC     0.0520
#> 2 normalize_t06mp 0.0100
#> 3 nzv_sUtzk       0.296

rec_ames |>
  extract_fit_time() |>
  count(part, wt = time, name = "time")
#> # A tibble: 2 × 2
#>   part    time
#>   <chr>  <dbl>
#> 1 bake  0.0320
#> 2 prep  0.326

[topepo]

Can we have a switch that will do the timing? I'd like to avoid the issue with diffs/reproducibility as we had with parsnip when we were tracking time.

[EmilHvitfeldt]

Can we have a switch that will do the timing? I'd like to avoid the issue with diffs/reproducibility as we had with parsnip when we were tracking time.

You mean an argument that will turn on/off the timing?

[simonpcouch]

🚀

[github-actions]

This pull request has been automatically locked. If you believe you have found a related problem, please file a new issue (with a reprex https://reprex.tidyverse.org) and link to this issue.