params

covid/python/covid.py

@@ -22,7 +22,7 @@ def timestep(S0, I0, R0, dt, beta, lamb, N):
 S0 = 400000
 I0 = 1
 R0 = 0
-max_t = 100
+max_t = 1000
 n_steps = 1000
 dt = max_t / n_steps
 
@@ -42,6 +42,8 @@ for i in range(0, n_steps):
     i_list.append(I1)
     r_list.append(R1)
     t_list.append((i+1)*dt)
+    # if ((i+1)*dt >= 10):
+    #     beta = 0.1, 0.25, 0.5
 
 s = np.array(s_list)
 r = np.array(r_list)