autoemulate.simulations.epidemic

autoemulate.simulations.epidemic#

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

Simulate an epidemic using the SIR model.

Parameters:
  • x (array-like) – 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, optional) – The total population size.

  • I0 (int, optional) – The initial number of infected individuals.

Returns:

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

Return type:

float