Skip to contents

Calculate a lagged difference

Source: R/step_lag_difference.R
step_lag_difference.Rd

step_lag_difference() creates a specification of a recipe step that will generate one or more new columns of derived data.

Usage

step_lag_difference(
  recipe,
  ...,
  role = "predictor",
  horizon = 7,
  prefix = "lag_diff_",
  skip = FALSE,
  id = rand_id("lag_diff")
)

Arguments

recipe

A recipe object. The step will be added to the sequence of operations for this recipe.

...

One or more selector functions to choose variables for this step. See recipes::selections() for more details.

role

For model terms created by this step, what analysis role should they be assigned? lag is default a predictor while ahead is an outcome.

horizon

Scalar or vector. Time period(s) over which to calculate differences.

prefix

A character string that will be prefixed to the new column.

skip

A logical. Should the step be skipped when the recipe is baked by bake()? While all operations are baked when prep() is run, some operations may not be able to be conducted on new data (e.g. processing the outcome variable(s)). Care should be taken when using skip = TRUE as it may affect the computations for subsequent operations.

id

A unique identifier for the step

Value

An updated version of recipe with the new step added to the sequence of any existing operations.

See also

Other row operation steps: step_epi_lag(), step_growth_rate()

Examples

r <- epi_recipe(case_death_rate_subset) %>%
  step_lag_difference(case_rate, death_rate, horizon = c(7, 14)) %>%
  step_epi_naomit()
r
#> 
#> ── Epi Recipe ──────────────────────────────────────────────────────────────────
#> 
#> ── Inputs 
#> Number of variables by role
#> raw:        2
#> geo_value:  1
#> time_value: 1
#> 
#> ── Operations 
#> 1. Calculating lag_difference for: case_rate and death_rate by 7, 14
#> 2. • Removing rows with NA values in: all_predictors()
#> 3. • Removing rows with NA values in: all_outcomes()

r %>%
  prep(case_death_rate_subset) %>%
  bake(case_death_rate_subset)
#> An `epi_df` object, 19,712 x 8 with metadata:
#> * geo_type  = state
#> * time_type = day
#> * as_of     = 2022-05-31 19:08:25.791826
#> 
#> # A tibble: 19,712 × 8
#>    geo_value time_value case_rate death_rate lag_diff_7_case_rate
#>  * <chr>     <date>         <dbl>      <dbl>                <dbl>
#>  1 ak        2021-01-14      37.4     0.0988               -4.07 
#>  2 al        2021-01-14      73.6     2.51                 -7.70 
#>  3 ar        2021-01-14      87.7     1.42                -10.3  
#>  4 as        2021-01-14       0       0                     0    
#>  5 az        2021-01-14     124.      2.14                  0.117
#>  6 ca        2021-01-14     108.      1.21                  2.37 
#>  7 co        2021-01-14      40.0     0.526                -6.33 
#>  8 ct        2021-01-14      75.4     1.07                  6.97 
#>  9 dc        2021-01-14      41.5     0.681                 6.07 
#> 10 de        2021-01-14      75.1     0.792                -5.53 
#> # ℹ 19,702 more rows
#> # ℹ 3 more variables: lag_diff_14_case_rate <dbl>, lag_diff_7_death_rate <dbl>,
#> #   lag_diff_14_death_rate <dbl>