epymorph.parameter_fitting.utils.data_loader

DataLoader

DataLoader(rume: RUME)

A class responsible for loading data from various sources such as simulations or CSV files.

Attributes:

• rume (Rume) –

  Simulation parameters and configuration.

Parameters:

• rume (Rume) –

  Simulation runtime environment or configuration object.

rume instance-attribute

rume = rume

load_data

load_data(
    observations: Observations,
) -> Tuple[ndarray, ndarray]

Loads the data from a given source.

This method handles two types of sources: 1. CSVTimeSeries: Data is loaded from a CSV file. 2. Any of the CDC adrios.

Parameters:

• observations (Observations) –

  The data source.

Returns:

• Tuple[ndarray, ndarray] –

  A tuple containing two numpy arrays: - dates: An array of date values. - sim_data: Simulated or observed case counts, generated using a Poisson distribution.

Raises:

• ValueError –

  Raised if the data source is not supported.