Skip to contents

Fit recent hospital admission and wastewater trends using BRMS

Usage

fit_recent_trends(
  forecast_date,
  location,
  scenario,
  raw_output_dir,
  hosp_prior_params,
  ww_prior_params,
  hosp_lookback_days,
  ww_lookback_days,
  seed,
  control,
  ...
)

Arguments

forecast_date

forecast_date for which to fits recent data trends. Data for an actual forecast must already have been produced and exist in raw_output_dir.

location

Location for which to fit trends.

scenario

Wastewater data availability scenario to analyze.

raw_output_dir

Directory containing raw output .rds files. Used to obtain the admissions and wastewater data used in fitting the forecasting model.

hosp_prior_params

Named list of values for prior hyperparameters needed by fit_hosp_trend().

ww_prior_params

Named list of values for prior hyperparameters needed by fit_ww_trend().

hosp_lookback_days

Number of days to look back when fitting the trend in hospital admissions.

ww_lookback_days

Number of days to look back when fitting the trend in wastewater concentrations.

seed

Seed for Stan's pseudorandom number generator.

control

list of control parameters passed to stan via brms::brm(). Default NULL.

...

Additional keyword arguments passed to brms::brm() via fit_hosp_trend().

Value

The two fit objects, as a list, saving raw fits to disk as a side effect.