如何从ARIMA模型打印AIC或BIC

时间:2016-07-13 01:24:23

标签: python regression statsmodels model-comparison

我已经创建了ARIMA模型,但我无法找到打印AIC或BIC结果的方法。我需要这些数字进行模型比较。不幸的是,sourceforge上的文档已经关闭,在查看statsmodel github存储库时我找不到我的答案。

这是我的代码:

import pandas as pd
import pandas.io.data
import statsmodels.formula.api as sm 
import matplotlib.pyplot as plt 
from statsmodels.tsa.arima_model import ARIMA

list = ['spy']
df = pd.io.data.get_data_yahoo(list, start = '2013-11-01', end = '2016-7-01', interval = 'm')['Adj Close'] 
df.dropna(inplace = True) 
df = df.pct_change()
df.dropna(inplace = True) 

model = ARIMA(df.spy, order = (0,0,1))
results_ARIMA = model.fit(disp=-1)
plt.plot(results_ARIMA.fittedvalues, color='red') 
plt.show() 

2 个答案:

答案 0 :(得分:1)

我在这里找到了解决方案。您需要从statsmodels.tsa.arima_model导入ARMAResults类。

from statsmodels.tsa.arima_model import ARMAResults 

一旦完成,您可以插入

print(ARMAResults.summary(results_ARIMA))

这将打印出包括BIC和AIC的结果摘要。

答案 1 :(得分:0)

results_ARIMA.aicresults_ARIMA.bic 会给你相应的值。您不需要括号,因为它们不可调用。