前提
Pythonを用いたネットワーク上でのSIRモデルを作成していました.
後述の"society.py"および"epidemics.py"については特にエラーは出なかったのですが,最後の"main.py"で以下のエラーが発生しました.
自分の考えではmodule 'society' に,属性 'generate_agents'は対応しているように思えるのですが,改善方法が分かりません.
実現したいこと
・エラーの解決
・プログラムが正常に作動するようにしたい
発生している問題・エラーメッセージ
AttributeError: module 'society' has no attribute 'generate_agents'
該当のソースコード1
society.py
import networkx as nx class Agent: """ state = ['S', 'I', 'R'] """ def __init__(self): self.state = 'S' self.neighbors_id = None def generate_agents(num_agent, average_degree): network = nx.barabashi_albert_graph(num_agent, average_degree//2) agents = [Agent() for agent_id in range(num_agent)] #エージェントのリスト for agent_id, agent in enumurate(agents): agent.nebghbors_id = list(network[agent_id]) return agents def count_state_fraction(agents): """ 各状態のエージェントの割合 """ fs = len([agent for agent in agents if agent.state == 'S'])/len(agents) fi = len([agent for agent in agents if agent.state == 'I'])/len(agents) fr = 1 - fs -fi return fs, fi, fr def count_num_i(agents): """ 感染状態のエージェントの数 """ num_i = len([agent for agent in agents if agent.state == 'I']) return num_i
該当のソースコード2
epidemics.py
import random as rnd import society def initialize_state(agents, num_initial_infected_agents=1): """ 初期保有者をランダムに決定 """ initial_infected_agent_id = rnd.sample(k = num_initial_infected_agent) for i, agent in enumurate(agents): if i in initial_infected_agent_id: agent.state = 'I' else: agent.state = 'S' def disease_spreading(agents, beta, gammma): """ 拡散ダイナミクス """ for time in range(1, 10000): state_changeable_agents = [agent for agent in agents if agent.state in ['S', 'I']] next_states = ['S' for i in range(len(state_changeable_agents))] for i, agent in enumurate(state_changeable_agents): if agent.state == 'S': num_infected_neighbors = len([agents[agent_id] for agent_id in agent.neighbors_id if agents[agent_id].state == 'I']) if rnd.random() <= beta*num_infected_neighbors: next_states[i] = 'I' else: pass elif agent.state == 'I': if rnd.random() <= gammma: next_states[i] = 'R' else: next_states[i] = 'I' # 状態の更新 for agent, next_state in zip(state_changeable_agents, next_states): agent.states = next_state fs, fi, fr = society.count_state_fraction(agents) num_i = society.count_num_i(state_changeable_agents) print(f'Time:{time}, Fs:{fs:2f}, Fi:{fi:2f}, Fr:{fr:.2f}') if num_i == 0: print('spreading finished') break return fs, fi, fr
該当のソースコード3
main.py
import numpy as np import pandas as pd import random as rnd import society import epidemics def main(): ### Calcualtion setting ### num_agent = 10000 # エージェントの総数 average_degree = 8 # 平均次数 beta = 0.14 # 感染率 gamma = 0.33 # 回復率 max_season = 1000 num_initial_infected_agents = 1 agents = society.generate_agents(num_agent, average_degree) result = pd.DataFrame({'FES':[]}) fes_hist = [] for season in range(1, max_season+1): fs, fim, fi, fr = epidemics.disease_spreading(agents, beta, gamma) fes_hist.append(fr) new_result = pd.DataFrame([fes_eq], columns = ['FES']) result = result.append(new_result) print(f'Season finished with FES: {fes_eq:.2f}') result.to_csv(f'result.csv') if __name__=='__main__': main()
試したこと
単純なスペルミスの確認
プロパティの確認
他のウェブサイトの調査
補足情報(FW/ツールのバージョンなど)
プログラムを作成するうえで参考にしたウェブサイト
https://qiita.com/Keyskey/items/8e0d7a67b222a7866154
ゲーム理論についての要素を省き,単純なSIRモデルとして動かしてみたいと考えています.
jupyter notebook

0 コメント