mirror of
https://github.com/gsi-upm/soil
synced 2024-11-14 23:42:29 +00:00
50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
|
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']
|