Я пытаюсь применить модель AR к данным о продажах ежедневных продаж 2018 года. При вызове функции прогнозирования для указанного ниже диапазона получается ключевая ошибка.
start_index = datetime(2018, 8, 29)
end_index = datetime(2018, 11, 28)
Error details
KeyError: 'only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indices'
фрагмент кода
start_index = datetime(2018, 8, 29)
end_index = datetime(2018, 11, 28)
AR_predict=AR_output.predict(start=start_index,end=end_index)
AR_predict=AR_predict.cumsum().shift().fillna(0)
AR_predict1=pd.Series(np.ones(test.shape[0]) * np.log(test['Volume'])[0], index = test.index)
AR_predict1=AR_predict1.add(AR_predict,fill_value=0)
AR_predict = np.exp(AR_predict1)
plt.plot(test['Volume'], label = "Test")
plt.plot(AR_predict, color = 'red', label = "Predict")
plt.legend(loc= 'best')
plt.title('RMSE: %.4f'% (np.sqrt(np.dot(AR_predict, test['Volume']))/test.shape[0]))
plt.show()
В качестве начальной попытки диапазон дат, который был передан, является диапазоном набора тестовых данных.Если он будет успешным, потребуется еженедельное и ежемесячное прогнозирование, начиная с декабря 2018 года.
Диапазон набора тестовых данных
start_index = datetime(2018, 8, 29)
end_index = datetime(2018, 11, 28)