Прогноз прямой линии для ARIMA - PullRequest
0 голосов
/ 25 апреля 2020

У меня есть среднечасовой трафик c данных с 16056 наблюдениями, пропущенных данных нет. введите описание изображения здесь

Я использовал ARIMA () из пакета statsmodels.tsa.arima_model. Из сюжета pacf и acf я получил модель ARIMA (2,1,1). Я пытаюсь предсказать почасовой трафик c на год. Но проблема, с которой я столкнулся, как вы можете видеть, заключается в том, что результат всегда получается прямой линией, а не зигзагом после прогноза.

Я что-то не так делаю?

# fit model
model = ARIMA(y_train, order=(2,1,1))
model_fit = model.fit(disp=0)

# predict
start_index = datetime(2012, 10, 15)
end_index = datetime(2012, 12, 31)
forecast = model_fit.predict(start=start_index, end=end_index)

# visualization
fig = plt.figure(figsize=(15, 4))
plt.plot(y_train.index, temp_df['total_users'], label='original')
plt.plot(forecast, label='predicted')
plt.title('Time Series Forecast')
plt.xlabel('Date')
plt.ylabel('Traffic')
plt.legend()
plt.show()
...