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

add fourth plot to present_model

Browse source
This commit is contained in:
markus 2021-02-03 17:24:11 +01:00
parent 4479cc07c0
commit 5a28d8602c
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
cara/montecarlo.py
View file

@ -365,6 +365,7 @@ def present_model(model: MCConcentrationModel, bins: int = 30) -> None:
fig.suptitle('Summary of model parameters')
plt.tight_layout()
plt.subplots_adjust(hspace=0.2)
plt.subplots_adjust(top=0.88)
fig.set_figheight(10)
for x, y in ((0, 0), (0, 1), (1, 0), (1, 1)):
@ -384,6 +385,13 @@ def present_model(model: MCConcentrationModel, bins: int = 30) -> None:
axs[0, 0].set_title('Viral load in sputum')
axs[0, 0].set_xlabel('Viral load [log10(RNA copies / mL)]')
ds = np.linspace(0.1, 30, 2000)
concentrations = model.infected._concentration_distribution()(ds)
axs[0, 1].plot(ds, concentrations)
axs[0, 1].set_title(r'Particle concentration vs diameter')
axs[0, 1].set_ylabel('Concentration [cm^-3]')
axs[0, 1].set_xlabel(r'Diameter [$\mu$m]')
categories = ("seated", "standing", "light exercise", "moderate exercise", "heavy exercise")
axs[1, 0].set_title(f'Breathing rate - '
f'{categories[model.infected.breathing_category - 1]}')