Update agents based on Bernoulli draws

update_agents(current_states, SIR_count, beta, gamma)

Arguments

current_states

vector of 0/1/2 of length n_agents. Gives the current state for each agent

SIR_count

total number of S, I, R currently in each state

beta

infection parameter between 0 and 1

gamma

infection parameter between 0 and 1

Value

list with

  • states vector of size n_agents of new states

  • SIR_count new total number of S, I, R currently in each state