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

add labels to pi_vs_viral_load plot

Browse source
This commit is contained in:
markus 2021-02-11 12:44:14 +01:00
parent 35247dd8f0
commit d78a46bc0b
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cara/montecarlo.py
View file

@ -531,6 +531,10 @@ def plot_pi_vs_viral_load(baseline: MCExposureModel, samples_per_vl: int = 20000
plt.plot(viral_loads, pi_medians)
plt.fill_between(viral_loads, lower_percentiles, upper_percentiles, alpha=0.2)
plt.title('Probability of infection vs viral load')
plt.ylabel('Percentage probability of infection')
plt.xticks(ticks=[i for i in range(3, 13)], labels=['$10^{' + str(i) + '}$' for i in range(3, 13)])
plt.xlabel('Viral load in sputum')
plt.show()