Separate models in modules

models/IndependentCascadeModel/IndependentCascadeModel.py

@@ -0,0 +1,56 @@
+import settings
+import random
+from ..BaseBehaviour import *
+from .. import sentimentCorrelationNodeArray
+
+settings.init()
+
+
+class IndependentCascadeModel(BaseBehaviour):
+    """
+    Settings:
+        innovation_prob
+
+        imitation_prob
+    """
+
+    def __init__(self, environment=None, agent_id=0, state=()):
+        super().__init__(environment=environment, agent_id=agent_id, state=state)
+        self.innovation_prob = settings.innovation_prob
+        self.imitation_prob = settings.imitation_prob
+        self.time_awareness = 0
+        sentimentCorrelationNodeArray[self.id][self.env.now]=0
+
+    def step(self,now):
+        self.behaviour()
+        super().step(now)
+
+    def behaviour(self):
+        aware_neighbors_1_time_step=[]
+        # Outside effects
+        if random.random() < settings.innovation_prob:
+            if self.state['id'] == 0:
+                self.state['id'] = 1
+                sentimentCorrelationNodeArray[self.id][self.env.now]=1
+                self.time_awareness = self.env.now #To know when they have been infected
+            else:
+                pass
+
+            self.attrs['status'] = self.state['id']
+            return
+
+        # Imitation effects
+        if self.state['id'] == 0:
+            aware_neighbors = self.get_neighboring_agents(state_id=1)
+            for x in aware_neighbors:
+                if x.time_awareness == (self.env.now-1):
+                    aware_neighbors_1_time_step.append(x)
+            num_neighbors_aware = len(aware_neighbors_1_time_step)
+            if random.random() < (settings.imitation_prob*num_neighbors_aware):
+                self.state['id'] = 1
+                sentimentCorrelationNodeArray[self.id][self.env.now]=1
+            else:
+                pass
+
+            self.attrs['status'] = self.state['id']
+            return

models/IndependentCascadeModel/__init__.py

@@ -0,0 +1 @@
+from .IndependentCascadeModel import IndependentCascadeModel