ARIMAモデルで予測の信頼区間を図示したい

Question

https://momonoki2017.blogspot.com/2018/03/python10.html
こちらのサイトを参考に、秒刻みで記録されたflow1のARIMAによる予測を行っています。

import statsmodels.api as sm
import pandas as pd
import numpy as np
import requests
import io
from matplotlib import pylab as plt
%matplotlib inline
# グラフを横長にする
from matplotlib.pylab import rcParams
rcParams['figure.figsize'] = 15, 6

row_data = pd.read_csv('2020.1.13sec.csv', encoding = "SHIFT-JIS")

# float型にしないとモデルを推定する際にエラーがでる
row_data.flow = row_data.flow.astype('float64')
row_data.flow1 = row_data.flow1.astype('float64')
# datetime型にしてインデックスにする
row_data.time = pd.to_datetime(row_data.time)
data = row_data.set_index("time")
# 線形補間
data['flow1'].interpolate(method='linear', inplace=True)

# 階差を取る
diff = data.flow1.diff()
diff = diff.dropna()

# Split data into train / test sets （約50000データあり）
train = data.iloc[:len(data)-10000] 
test = data.iloc[len(data)-10000:]

from statsmodels.tsa.arima_model import ARIMA
model=ARIMA(train['flow1'],order=(5,1,2))
results=model.fit()

start=len(train)
end=len(train)+len(test)-1
pred=results.predict(start=start,end=end,typ='levels').rename('ARIMA Predictions')
pred.plot(legend=True)
test['flow1'].plot(legend=True)

predict_dy = results.get_prediction(start ='2021/1/14 3:41:01', end='2021/1/14 6:27:40')
type(predict_dy)

AttributeError Traceback (most recent call last)

<ipython-input-50-d23d818922c4> in <module>
----> 1 predict_dy = results.get_prediction(start ='2021/1/14 3:41:01', end='2021/1/14 6:27:40')
2 type(predict_dy)
~/opt/anaconda3/lib/python3.8/site-packages/statsmodels/base/wrapper.py in getattribute(self, attr)
34 pass
35
---> 36 obj = getattr(results, attr)
37 data = results.model.data
38 how = self._wrap_attrs.get(attr)
AttributeError: 'ARIMAResults' object has no attribute 'get_prediction'

ここで行き詰まってしまいます。
最終的には

plt.figure(figsize=(8, 4))

plt.plot(test, label="Actual")
plt.plot(predict_dy.predicted_mean, c="b", label="model-pred", alpha=0.7)

plt.fill_between(predict_dy_ci.index, predict_dy_ci.iloc[:, 0], predict_dy_ci.iloc[:, 1], color='g', alpha=0.2)
plt.legend(loc='upper left')

を用いて予測の信頼区間を塗りつぶした図を目指しています。
get_predictionがうまくいかない理由がわかりません。

Answer 1

arima_modelじゃなくてarima.modelにはget_prediction()があるみたいです
statsmodels.tsa.arima.model.ARIMAResults.get_prediction

Comments

[multinguish]

無事解決しました。
本当にありがとうございました。