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']