⬆️ SEIR/SEIS discrete & continuous time

GiulioRossetti · Jul 21, 2020 · 8d0a5cc · 8d0a5cc
1 parent 1e4780c
commit 8d0a5cc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ndlib/models/epidemics/SEIR_ct_Model.py b/ndlib/models/epidemics/SEIR_ct_Model.py
@@ -53,7 +53,7 @@ def iteration(self, node_status=True):
         self.clean_initial_status(self.available_statuses.values())
 
         actual_status = {node: nstatus for node, nstatus in future.utils.iteritems(self.status)}
-        self.progress_I = {node: 1 for node in actual_status if actual_status[node] == 1}
+        self.progress_I = {node: 0 for node in actual_status if actual_status[node] == 1}
 
         if self.actual_iteration == 0:
             self.actual_iteration += 1

diff --git a/ndlib/models/epidemics/SEIS_ct_Model.py b/ndlib/models/epidemics/SEIS_ct_Model.py
@@ -66,7 +66,7 @@ def iteration(self, node_status=True):
         self.clean_initial_status(self.available_statuses.values())
 
         actual_status = {node: nstatus for node, nstatus in future.utils.iteritems(self.status)}
-        self.progress_I = {node: 1 for node in actual_status if actual_status[node] == 1}
+        self.progress_I = {node: 0 for node in actual_status if actual_status[node] == 1}
 
         if self.actual_iteration == 0:
             self.actual_iteration += 1