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