From 9b95dca2fbfe566efc1f9d0aeca3055731048e7e Mon Sep 17 00:00:00 2001
From: Mahmoud Bentriou <mahmoud.bentriou@centralesupelec.fr>
Date: Wed, 30 Jun 2021 16:17:12 +0200
Subject: [PATCH] small fix in sir tau leap simulation
---
models/SIR_tauleap.jl | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/models/SIR_tauleap.jl b/models/SIR_tauleap.jl
index e6743b5..ec676ab 100644
--- a/models/SIR_tauleap.jl
+++ b/models/SIR_tauleap.jl
@@ -1,13 +1,13 @@
@everywhere import Distributions: Poisson, rand
-d=3
+d=4
k=3
-dict_var_SIR_tauleap = Dict(:S => 1, :I => 2, :R => 3)
+dict_var_SIR_tauleap = Dict(:S => 1, :I => 2, :R => 3, :step => 4)
dict_p_SIR_tauleap = Dict(:ki => 1, :kr => 2, :tau => 3)
l_tr_SIR_tauleap = [:U]
p_SIR_tauleap = [0.0012, 0.05, 5.0]
-x0_SIR_tauleap = [95, 5, 0]
+x0_SIR_tauleap = [95, 5, 0, 0]
t0_SIR_tauleap = 0.0
@everywhere function SIRTauleap_f!(xnplus1::Vector{Int}, l_t::Vector{Float64}, l_tr::Vector{Transition},
xn::Vector{Int}, tn::Float64, p::Vector{Float64})
@@ -28,6 +28,7 @@ t0_SIR_tauleap = 0.0
for i = 1:3
@inbounds xnplus1[i] = xn[i]+ nbr_R1*nu_1[i] + nbr_R2*nu_2[i]
end
+ xnplus1[4] += 1
l_t[1] = tn + tau
l_tr[1] = :U
end
--
GitLab