diff --git a/models/SIR_tauleap.jl b/models/SIR_tauleap.jl index e6743b5d27f0935eef71f63af27e3d28f9424b0b..ec676abca8b29c8eec8e6db57705dc0fd117580b 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