Separate models in modules

models/BassModel/BassModel.py

@@ -0,0 +1,49 @@
+import settings
+import random
+from ..BaseBehaviour import *
+from .. import sentimentCorrelationNodeArray
+
+settings.init()
+
+
+class BassModel(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
+        sentimentCorrelationNodeArray[self.id][self.env.now]=0
+
+    def step(self, now):
+        self.behaviour()
+        super().step(now)
+
+    def behaviour(self):
+        # Outside effects
+        if random.random() < settings.innovation_prob:
+            if self.state['id'] == 0:
+                self.state['id'] = 1
+                sentimentCorrelationNodeArray[self.id][self.env.now]=1
+            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)
+            num_neighbors_aware = len(aware_neighbors)
+            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']

models/BassModel/__init__.py

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