add print_qd_info

This commit is contained in:
markus 2021-03-01 14:44:52 +01:00
parent e9621b7815
commit bad2ba9c6a
2 changed files with 23 additions and 8 deletions

View file

@ -11,6 +11,7 @@ from cara.model_scenarios import *
#print(np.quantile(chorale_model.infection_probability(),0.90))
#print(np.quantile(chorale_model.infection_probability(),0.1))
print_qd_info(large_population_baselines[0])
#print(np.mean(shared_office_model[1].infection_probability()))
@ -31,12 +32,12 @@ from cara.model_scenarios import *
# 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'
)
# 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='')

View file

@ -538,7 +538,7 @@ def plot_pi_vs_viral_load(baselines: typing.Union[MCExposureModel, typing.List[M
number=infected.number,
presence=infected.presence,
masked=infected.masked,
expiratory_activity=infected.expiratory_activity,
expiratory_activity=infected.exquantapiratory_activity,
breathing_category=infected.breathing_category,
virus=infected.virus,
samples=samples_per_vl,
@ -1094,4 +1094,18 @@ def compare_concentration_curves(exp_models: typing.List[MCExposureModel], label
plt.tight_layout()
plt.hlines([60], colors=['lightgrey'], linestyles=['dashed'], xmin=start, xmax=stop)
plt.text(7, 65, "$qID = 60$", color='lightgrey')
plt.show()
plt.show()
def print_qd_info(model: MCExposureModel) -> None:
qds = model.exposed.activity.inhalation_rate * (1 - model.exposed.mask.η_inhale) * model.quanta_exposure() * 0.6
print(f"----- qD distribution -----\n"
f"Mean:\t{np.mean(qds)}\n"
f"Median:\t{np.median(qds)}\n\n"
f"Percentiles\n"
f"1st:\t{np.percentile(qds, 1)}\n"
f"5th:\t{np.percentile(qds, 5)}\n"
f"10th:\t{np.percentile(qds, 10)}\n"
f"90th:\t{np.percentile(qds, 90)}\n"
f"95th:\t{np.percentile(qds, 95)}\n"
f"99th:\t{np.percentile(qds, 99)}\n")