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

add labels to compare_infection_probabilities_vs_viral_loads

Browse source
This commit is contained in:
markus 2021-02-16 14:38:39 +01:00
parent a3dffcf1e7
commit 75225e88c7
1 changed files with 10 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
cara/montecarlo.py
View file

@ -726,10 +726,20 @@ def compare_infection_probabilities_vs_viral_loads(baseline1: MCExposureModel, b
mean_ratios.append(p1_mean / p2_mean)
plt.plot(viral_loads, mean_ratios)
plt.ylabel(f"Ratio of mean P(i) values - P(i|qID = {baseline1.concentration_model.infected.qid}) / P(i|qID = {baseline2.concentration_model.infected.qid})")
plt.xlabel("Viral load in sputum")
plt.xticks(ticks=[i for i in range(3, 11)], labels=['$10^{' + str(i) + '}$' for i in range(3, 11)])
plt.show()
plt.plot(viral_loads, p1_means)
plt.plot(viral_loads, p2_means)
plt.ylim(0, 100)
plt.ylabel("Percentage probability of infection")
plt.xlabel("Viral load in sputum")
plt.legend([f'qID = {baseline1.concentration_model.infected.qid}',
f'qID = {baseline2.concentration_model.infected.qid}'])
plt.xticks(ticks=[i for i in range(3, 11)], labels=['$10^{' + str(i) + '}$' for i in range(3, 11)])
plt.show()