diff --git a/cara/mc-output.py b/cara/mc-output.py index ce39facf..e31b7d19 100644 --- a/cara/mc-output.py +++ b/cara/mc-output.py @@ -1,9 +1,11 @@ from cara.montecarlo import * from cara.model_scenarios import * +#plot_concentration_curve(classroom_model[1]) #compare_concentration_curves([classroom_model, classroom_model_with_hepa], ['Just window', 'Window and HEPA']) -#print(np.mean(chorale_model.infection_probability())) +#print(np.mean(classroom_model[0].infection_probability())) +#print(np.mean(classroom_model[1].infection_probability())) #print(np.mean(chorale_model.infection_probability())+np.std(chorale_model.infection_probability())) #print(np.quantile(chorale_model.infection_probability(),0.8)) #print(np.quantile(chorale_model.infection_probability(),0.90)) @@ -11,17 +13,31 @@ from cara.model_scenarios import * -print(np.mean(shared_office_model[1].infection_probability())) -composite_plot_pi_vs_viral_load([shared_office_worst_model[1], shared_office_model[1], shared_office_better_model[1]], - labels=['No mask &\nwindows closed', 'Baseline', 'Baseline +\nHEPA filter'], - colors=['tomato', '#1f77b4', 'limegreen'], - title='$P(I|qID)$ vs $vl$ - Shared office scenario', - vl_points=200) -composite_plot_pi_vs_viral_load([classroom_model[1], classroom_model_with_hepa[1]], - labels=['No mask &\nwindows closed', 'Baseline', 'Baseline +\nHEPA filter'], - colors=['tomato', '#1f77b4', 'limegreen'], - title='$P(I|qID)$ vs $vl$ - Shared office scenario', - vl_points=200) +#print(np.mean(shared_office_model[1].infection_probability())) +#composite_plot_pi_vs_viral_load([shared_office_worst_model[1], shared_office_model[1], shared_office_better_model[1]], +# labels=['No mask &\nwindows closed', 'Baseline', 'Baseline +\nHEPA filter'], +# colors=['tomato', '#1f77b4', 'limegreen'], +# title='$P(I|qID)$ vs $vl$ - Shared office scenario', +# vl_points=200) +#composite_plot_pi_vs_viral_load([classroom_model_no_vent[1], classroom_model[1], classroom_model_with_hepa[1], classroom_model_full_open_hepa[1]], +# labels=['Windows closed', 'Baseline:\n(windows 10min/2h)', 'Baseline:\n(windows 10min/2h)+\nHEPA', 'Multiple windows open +\nHEPA filter'], +# colors=['tomato','#1f77b4', 'seagreen', 'limegreen'], +# title='', +# vl_points=200) + +#compare_concentration_curves([classroom_model_no_vent[1], classroom_model[1], classroom_model_with_hepa[1], classroom_model_full_open_hepa[1]], +# labels=['Windows closed', 'Baseline:(windows 10min/2h)', 'Baseline:(windows 10min/2h) + HEPA', 'Multiple windows open + HEPA filter'], +# colors=['tomato','#1f77b4', 'seagreen', 'limegreen'], +# title='Classroom scenario' +# ) + +compare_concentration_curves([waiting_room_model[1], waiting_room_model_full_summer[1], + waiting_room_model_full_winter[1], waiting_room_model_periodic_winter[1]], + labels=['Baseline:(windows closed)', 'Windows open (summer)', 'Windows open (winter)', 'Windows open 5min/20min (winter)'], + colors=['#1f77b4', 'darkorange', 'deepskyblue', 'lightskyblue'], + title='Waiting room scenario' + ) + #plot_pi_vs_viral_load([shared_office_model[1]], labels=['Baseline'],title='')