def SEIR_sim(beta, gamma, alpha, E0, I0, days, step=0.1): x0 = vect([1.0 - E0 - I0, E0, I0, 0.0]) return euler_method(SEIR_model(beta, gamma, alpha), 0, x0, days, step)
def f(t, x): S, E, I, R = x return vect([ -beta * S * I, beta * S * I - alpha * E, alpha * E - gamma * I, gamma * I ])