u++の備忘録

Facebookの時系列解析ライブラリ「Prophet」で日経平均株価の終値を予測

データは、以下の記事と同様に取得しました(2007年〜2017年9月8日)。
upura.hatenablog.com

今回は下記の記事に沿って、デフォルト設定で処理してみました。
d.hatena.ne.jp

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import pandas as pd
from matplotlib import pyplot as plt

df = pd.read_csv('nikkei_index.csv')
plt.plot(df['y'])

from fbprophet import Prophet
model = Prophet()
model.fit(df)
future_df = model.make_future_dataframe(365)

forecast_df = model.predict(future_df)
model.plot(forecast_df)
plt.show()

実データ

f:id:upura:20171025175325p:plain

365日分の予測結果を追記

f:id:upura:20171025175331p:plain