noah/CS-7863-Sci-Stat-Final
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Breaking I out into D and R subgroups

Browse Source
This commit is contained in:
Noah L. Schrick 2023-04-28 18:54:53 -05:00
parent 7942bbb48a
commit 23e3096863
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
Analysis/prep_model.py
View File

@ -24,7 +24,8 @@ node_pos = {} # used for drawing/graphing
# Compartments # Compartments
S = 0 S = 0
I = 0 I_R = 0
I_D = 0
E = 0 E = 0
R = 0 R = 0
D = 0 D = 0
@ -68,30 +69,33 @@ for node in A:
else: else:
color_map.append(color) color_map.append(color)
color_d[node] = color color_d[node] = color
I = I + 1
# Check if node dies # Check if node dies
out_edges = list(G.out_edges(node)) out_edges = list(G.out_edges(node))
if not out_edges: if not out_edges:
D = D + 1 D = D + 1
I_D = I_D + 1
else:
I_R = I_R + 1
# Check if imported # Check if imported
in_edges = list(G.in_edges(node)) in_edges = list(G.in_edges(node))
if not in_edges: if not in_edges:
ep_tmp = ep_tmp + 1 ep_tmp = ep_tmp + 1
# Params # Params
beta = I/len(A) # rate of infec (I/total?) beta = (I_R+I_D)/len(A) # rate of infec (I/total?)
delta = E/len(A) # symptom appearance rate (E/total?) delta = E/len(A) # symptom appearance rate (E/total?)
gamma_r = R/len(A) # recov rate (R/total?) gamma_r = R/len(A) # recov rate (R/total?)
gamma_d = D/len(A) # death rate (D/total?) gamma_d = D/len(A) # death rate (D/total?)
mu = D/I # fatality ratio (D/I) mu = D/(I_R+I_D) # fatality ratio (D/I)
epsilon = ep_tmp/len(A) # infected import rate epsilon = ep_tmp/len(A) # infected import rate
omega = 0 # waning immunity rate omega = 0 # waning immunity rate
print("Model Compartments:") print("Model Compartments:")
print("S:", str(S)) print("S:", str(S))
print("I:", str(I))
print("E:", str(E)) print("E:", str(E))
print("I_R:", str(I_R))
print("I_D:", str(I_D))
print("R:", str(R)) print("R:", str(R))
print("D:", str(D)) print("D:", str(D))
print("\n") print("\n")