Как прогнозировать будущие даты, используя прогнозирование временных рядов в Python?

Question

Я новичок в прогнозировании временных рядов и создал следующую модель:

df = pd.read_csv('timeseries_data.csv', index_col="Month")

# ARMA
from statsmodels.tsa.arima_model import ARMA
from random import random
# contrived dataset
data = df
# fit model
model = ARMA(data, order=(2, 1))
model_fit = model.fit(disp=False)
# make prediction
yhat = model_fit.predict()
print(yhat)

Приведенный выше код прогнозирует точки данных в df. Но что мне делать, если я хочу предсказать какую-то будущую дату? Например: за весь 2022 год?

Answer 1

Я нашел решение, оно было на самом деле очень простым:

arma_df = model_fit.predict(start='2020-01-01', end='2023-12-01')