def case_age_model():
group = [
['S_kids', 'S_normal', 'S_risk'],
['E_kids', 'E_normal', 'E_risk'],
['I_kids', 'I_normal', 'I_risk'],
['I2'],
['I3'],
['R'],
['D']
]
label = {
"S_kids": "S",
"E_kids": "E",
"I_kids": "I_1",
"I2": "I_2",
"I3": "I_3"
}
data = get_results(
simulation_age_model, (N, B_KIDS, G_KIDS, P_KIDS, B_NORMAL, G_NORMAL,
P_NORMAL, B_NORMAL, G_NORMAL, P_NORMAL), NUM_SIM
)
times, avg, std = fill_around_std(
kolor_palette, data, 1, states=group, labels=label
)
save_data([times, avg, std], (N, NUM_SIM,))
def case_no_intervention():
label = {
"I1": "I_1",
"I2": "I_2",
"I3": "I_3"
}
data = get_results(simulation_no_intervention, (N,), NUM_SIM)
times, avg, std = fill_around_std(color_palette, data, 1, labels=label)
save_data([times, avg, std], (N, NUM_SIM))
def case_average_second_wave():
data = get_results(simulation_second_wave, tuple(), NUM_SIM)
labels = {
"I1": "I_{1}", "I2": "I_{2}", "I3": "I_{3}"
}
times, avg, std = fill_around_std(
color_palette, data, 1, labels=labels
)
save_data([times, avg, std], (N, NUM_SIM))
def case_average_time():
t = 5
output = get_results(
simulation_max_time, (N, t), NUM_SIM
)
timepoints, avg, std = fill_around_std(
color_palette, output, 1
)
save_data([timepoints, avg, std], (N, NUM_SIM, t))
def contact_tracing(p):
time = [5, 7.5, 10]
data = get_results(simulation_contact_tracing, (N, p, time), NUM_SIM)
states = [
["S"], ["E"], ["I1"], ["I2"], ["I3"], ["R"], ["D"], ["Q1", "Q2", "Q3"],
["QS", "QE"]
]
labels = {
"Q1": "Q_I", "QS": "Q_{S, E}",
"I1": "I_1", "I2": "I_2", "I3": "I_3"
}
times, avg, std = fill_around_std(
quolor_palette, data, 1, states=states, labels=labels
)
save_data([times, avg, std], (N, NUM_SIM, time, p))
def case_average_quarantine_rates():
q1 = 0.5
output = get_results(
simulation_max_quarantine_rates, (N, q1, 0, 0), NUM_SIM
)
labels = {
"I1": "I_1", "I2": "I_2", "I3": "I_3",
"Q1": "Q", "Q2": "Q_2", "Q3": "Q_3",
}
groups = [
["Q1", "Q2", "Q3"]
]
complete_group(groups, qolor_palette)
timepoints, avg, std = fill_around_std(
qolor_palette, output, 1, labels=labels, states=groups
)
save_data([timepoints, avg, std], (N, NUM_SIM, q1))
def case_infectious_tracing():
time = [5, 7.5, 10]
p = [0.7, 1]
data = get_results(
simulation_infectious_tracing, (N, p[0], p[1], time),
NUM_SIM,
)
states = [
["S"], ["E"], ["I1"], ["I2"], ["I3"], ["R"], ["D"], ["Q1", "Q2", "Q3"],
["QS", "QE"]
]
labels = {
"Q1": "Q_{I}", "QS": "Q_{S, E}",
"I1": "I_{1}", "I2": "I_{2}", "I3": "I_{3}"
}
times, avg, std = fill_around_std(
quolor_palette, data, 1, states=states, labels=labels
)
save_data([times, avg, std], (N, NUM_SIM, time, p))