prophet で株価予測メモ

FB社が公開している時系列予測ツール prophet を使うと、今後の株価の動きが予測できるそうです。使い方のメモ。　Google Colaboratoryで実行できます。

インストルール

!pip install prophet

株価データの準備

stooq から取得

import pandas_datareader.data as pdr

df = pdr.DataReader("{}.JP".format(2929), "stooq").sort_index()
df.tail()

分析用に加工

prophet で分析するために ds:日付, y:終値のカラムを設ける

df["ds"] = df.index
df = df.rename(columns={'Close': 'y'})
df.tail()

分析実行

from prophet import Prophet

m = Prophet()
m.fit(df)
future = m.make_future_dataframe(periods=365)
forecast = m.predict(future)
fig = m.plot(forecast)

2021-09-10 以降のグラフが予想結果