超簡単Pythonで株価予測（Prophet 利用）時系列予測

PythonでProphetを利用して25日先までの株価予測を超簡単に時系列予測（Facebook製）

Amazon製の同様ツールについては過去の投稿をどうぞ

1. ツールインストール

＄ pip install matplotlib fbprophet pandas-datareader

2. ファイル作成

pred.py

import pandas as pd
from fbprophet import Prophet
import pandas_datareader as pdr
import matplotlib.pyplot as plt

df = pdr.get_data_yahoo("AAPL", "2019-11-01", "2020-11-01")
df = pd.DataFrame({"ds": df.index.values, "y": df["Close"].values})
m = Prophet()
m.fit(df)
# 25日先まで予測
future = m.make_future_dataframe(periods=25, freq="d")
# 土日除外
future = future[future["ds"].dt.weekday < 5]
forecast = m.predict(future)
m.plot(forecast)
plt.savefig("pred.png")

3. 実行

$ python pred.py

以上、超簡単！

4. 参考

Quick Start