Skip to contents

Unified NA omission wrapper function for recipes

Usage

step_epi_naomit(recipe)

Arguments

recipe

Recipe to be used for omission steps

Value

Omits NA's from both predictors and outcomes at training time to fit the model. Also only omits associated predictors and not outcomes at prediction time due to lack of response and avoidance of data loss.

Examples

case_death_rate_subset %>%
  epi_recipe() %>%
  step_epi_naomit()
#> 
#> ── Epi Recipe ──────────────────────────────────────────────────────────────────
#> 
#> ── Inputs 
#> Number of variables by role
#> raw:        2
#> geo_value:  1
#> time_value: 1
#> 
#> ── Operations 
#> 1.  Removing rows with NA values in: all_predictors()
#> 2.  Removing rows with NA values in: all_outcomes()