autoemulate.simulations.epidemic

autoemulate.simulations.epidemic#

class Epidemic(parameters_range=None, output_names=None, log_level='progress_bar')[source]#

Bases: Simulator

Simulator of infectious disease spread (SIR).

simulate_epidemic(x, N=1000, I0=1)[source]#

Simulate an epidemic using the SIR model.

Parameters:
  • x (NumpyLike) – The parameters of the SIR model. The first element is the transmission rate (beta) and the second element is the recovery rate (gamma).

  • N (int) – The total population size. Defaults to 1000.

  • I0 (int) – The initial number of infected individuals. Defaults to 1.

Returns:

peak_infection_rate – The peak infection rate as a fraction of the total population.

Return type:

float