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

add time_step and fix shaded area

Browse source
This commit is contained in:
markus 2021-02-22 17:10:46 +01:00
parent 5c9181d798
commit 4467f36a7c
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
cara/montecarlo.py
View file

@ -12,7 +12,7 @@ import matplotlib.patches as patches
import matplotlib.lines as mlines
from sklearn.neighbors import KernelDensity
# This is a test comment
TIME_STEP = 0.01
USE_SCOEH = False
@ -358,7 +358,7 @@ class MCExposureModel:
exposure = np.zeros(self.concentration_model.infected.samples)
for start, stop in self.exposed.presence.boundaries():
times = np.arange(start, stop, 0.05)
times = np.arange(start, stop, TIME_STEP)
concentrations = np.asarray([self.concentration_model.concentration(t) for t in times])
integrals = np.trapz(concentrations, times, axis=0)
exposure += integrals
@ -854,7 +854,7 @@ def plot_concentration_curve(model: MCExposureModel):
transitions = model.concentration_model.infected.presence.transition_times()
start, stop = min(transitions), max(transitions)
times = np.arange(start, stop, 0.05)
times = np.arange(start, stop, TIME_STEP)
concentrations = [model.concentration_model.concentration(t) for t in tqdm(times)]
@ -870,7 +870,7 @@ def plot_concentration_curve(model: MCExposureModel):
# Plot presence of exposed person
for i, (presence_start, presence_finish) in enumerate(model.exposed.presence.boundaries()):
plt.fill_between(
times, np.max(upper) * 1.1, 0,
times, upper, 0,
where=(np.array(times) > presence_start) & (np.array(times) < presence_finish),
color=color, alpha=0.1,
label="Presence of exposed person(s)" if i == 0 else ""