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
])