pingu_98/cara
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add vlcrit values

Browse source
This commit is contained in:
markus 2021-02-24 10:40:14 +01:00
parent 758afafbcf
commit 04dfa59fce
2 changed files with 12 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
cara/mc-output.py
View file

@ -15,7 +15,8 @@ from cara.model_scenarios import *
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')
title='$P(I|qID)$ vs $vl$ - Shared office scenario',
vl_points=200)
#plot_pi_vs_viral_load([shared_office_model[1]], labels=['Baseline, qID=60', 'HEPA, qID=60', 'No mask + windows closed, qID=60'],title='$P(I|qID)$ - Shared office scenario')

14
cara/montecarlo.py
View file

@ -687,15 +687,21 @@ def composite_plot_pi_vs_viral_load(baselines: typing.List[MCExposureModel], lab
#axs[0, 2].text(axs[0, 2].get_xlim()[1] * 0.1, -0.05, '$(iii)$')
axs[0, 2].set_title('$(iii)$', fontsize=10)
axs[0, 0].text(2.5, 0.4, '$vl_{crit2}=$', fontsize=10, color='tomato')
axs[0, 0].text(2.5, 0.3, '$vl_{crit2}=$', fontsize=10, color='#1f77b4')
axs[0, 0].text(2.5, 0.2, '$vl_{crit2}=$', fontsize=10, color='limegreen')
crits = []
for line in lines:
for i, point in enumerate(line):
if point >= 0.95:
crits.append(viral_loads[i])
break
for i, (crit, color) in enumerate(zip(crits, colors)):
axs[0, 0].text(2.5, 0.4 - i * 0.1, f'$vl_{"{crit2}"}=' + '10^{' + str(np.round(crits[i], 1)) + '}$', fontsize=10, color=color)
if show_lines:
middle_positions = []
for line in lines:
for i, point in enumerate(line):
if point > 0.5:
if point >= 0.5:
middle_positions.append(viral_loads[i])
break