Datasets

Datasets#

Generation interval#

load_generation_interval()[source]#

Load the generation interval dataset

This dataset contains the generation interval distribution for COVID-19.

Returns:

The generation interval dataset

Return type:

pl.DataFrame

Notes

This dataset was downloaded directly from: https://raw.githubusercontent.com/CDCgov/wastewater-informed-covid-forecasting/0962c5d1652787479ac72caebf076ab55fe4e10c/input/saved_pmfs/generation_interval.csv

The dataset contains the following columns:

  • timepoint

  • probability_mass

Infection to admission interval#

load_infection_admission_interval()[source]#

Load the infection to admission interval

This dataset contains the infection to admission interval distribution for COVID-19.

Returns:

The infection to admission interval dataset

Return type:

pl.DataFrame

Notes

This dataset was downloaded directly from: https://raw.githubusercontent.com/CDCgov/wastewater-informed-covid-forecasting/0962c5d1652787479ac72caebf076ab55fe4e10c/input/saved_pmfs/inf_to_hosp.csv

The dataset contains the following columns:

  • timepoint

  • probability_mass

Simulated Wastewater Data#

This module loads the package dataset named ‘wastewater’ and provides functions to manipulate the data. It uses the ‘polars’ library.

load_wastewater()[source]#

Load the wastewater dataset. This dataset contains simulated entries of COVID-19 wastewater concentration data. The dataset is used to demonstrate the use of the wastewater-informed COVID-19 forecasting model.

Returns:

The wastewater dataset.

Return type:

pl.DataFrame

Notes

This dataset was downloaded directly from: CDCgov/wastewater-informed-covid-forecasting

The dataset contains the following columns:

  • lab_wwtp_unique_id

  • log_conc

  • date

  • lod_sewage

  • below_lod

  • daily_hosp_admits

  • daily_hosp_admits_for_eval

  • pop

  • forecast_date

  • hosp_calibration_time

  • site

  • ww_pop

  • inf_per_capita