ARIMA 模型在新的 statsmodels ARIMA for python 中无法正常工作

Question

我以前用过

from statsmodels.tsa.arima_model import ARIMA
model = ARIMA(log_air_passengers, order=(2, 1, 0))  
results_AR = model.fit(disp=-1)  
plt.plot(log_air_passengers_diff)
plt.plot(results_AR.fittedvalues, color='red')
plt.title('RSS: %.4f'% sum((results_AR.fittedvalues-log_air_passengers_diff)**2))

我会得到的情节是：

但现在他们有了更新版本的 ARIMA，并且在 0.12 发布后旧版本将被删除。所以，我正在尝试更新的：

from statsmodels.tsa.arima.model import ARIMA as ARIMA2
model = ARIMA2(log_air_passengers, order=(2, 1, 0), missing='drop')  
results_AR = model.fit()  
plt.plot(log_air_passengers_diff)
plt.plot(results_AR.fittedvalues, color='red')
plt.title('RSS: %.4f'% sum((results_AR.fittedvalues-log_air_passengers_diff)**2))

而且情节也很糟糕： 但它不起作用。我试图摆弄它，但它没有多大帮助。如何更改代码以使其正常工作？

Answer 1

你应该使用 SARIMAX

from statsmodels.tsa.statespace.sarimax import SARIMAX

model = SARIMAX(log_air_passengers,order=(1,1,0),simple_differencing=True)
result_AR = model.fit()
plt.plot(log_air_passengers_diff)
plt.plot(result_AR.fittedvalues,color='red')
plt.title('RSS: %.4f'%sum((result_AR.fittedvalues[1:] - log_air_passengers_diff)**2))