[python]prophet岡山県新型コロナ新規陽性者数を予測してみた（2022/12/15データ）

pip install prophet
pip install plotly

import pandas as pd
import matplotlib.pyplot as plt
from prophet import Prophet
from prophet.plot import plot_plotly,plot_components_plotly
from tkinter import filedialog

#ファイルダイアログのデフォルトフォルダはinitialdirで指示してください
filename=filedialog.askopenfilename(initialdir='C:/okayama-covid19-predict/',filetypes=[("csv file",".csv")])

#厚生労働省オープンデータ　https://www.mhlw.go.jp/stf/covid-19/open-data.html

okayama_popu=19000 #岡山県の人口の1/100

data=pd.read_csv(filename,index_col='Date')

# Prophet投入用インプットデータ作成
df = pd.DataFrame({"ds":data.index, "y":data["Okayama"]}).reset_index(drop=True)

# モデルインスタンス
#proph = Prophet() #線形トレンド　デフォルト
proph = Prophet(growth='logistic')

#非線形トレンドはキャパシティ設定必要
df['cap']=okayama_popu

# モデル学習
proph.fit(df)

# 未来予測用のデータフレーム
future = proph.make_future_dataframe(periods=365)

#未来予測用データフレームにもキャパシティ設定
future['cap']=okayama_popu

# 時系列を予測
forecast = proph.predict(future)

figure = plot_plotly(proph,      # 時系列モデル
                     forecast,   # 予測結果
                    )

figure2=proph.plot_components(forecast)

proph.plot(forecast)
plot_components_plotly(proph,forecast)

# 出力
plt.show()