epi_workflow: Create an epi_workflow

In cmu-delphi/epipredict: Basic epidemiology forecasting methods

Create an epi_workflow

Description

This is a container object that unifies preprocessing, fitting, prediction, and postprocessing for predictive modeling on epidemiological data. It extends the functionality of a workflows::workflow() to handle the typical panel data structures found in this field. This extension is handled completely internally, and should be invisible to the user. For all intents and purposes, this operates exactly like a workflows::workflow(). For more details and numerous examples, see there.

Usage

epi_workflow(preprocessor = NULL, spec = NULL, postprocessor = NULL)

Arguments

preprocessor	An optional preprocessor to add to the workflow. One of: A formula, passed on to add_formula(). A recipe, passed on to add_recipe(). A workflow_variables() object, passed on to add_variables().
spec	An optional parsnip model specification to add to the workflow. Passed on to add_model().
postprocessor	An optional postprocessor to add to the workflow. Currently only frosting is allowed using, add_frosting().

Value

A new epi_workflow object.

See Also

workflows::workflow

Examples

jhu <- covid_case_death_rates

r <- epi_recipe(jhu) %>%
  step_epi_lag(death_rate, lag = c(0, 7, 14)) %>%
  step_epi_ahead(death_rate, ahead = 7) %>%
  step_epi_lag(case_rate, lag = c(0, 7, 14)) %>%
  step_epi_naomit()

wf <- epi_workflow(r, parsnip::linear_reg())

wf

cmu-delphi/epipredict documentation built on March 5, 2025, 12:17 p.m.