text_html_output_2.png text_html_output_1.png
from math import sqrt

from sklearn.metrics import mean_squared_error

from statsmodels.tsa.arima.model import ARIMA

model = ARIMA(train, order=(1, 1, 2))
model_fit = model.fit()

predictions = model_fit.forecast(steps=len(test))[0]

rmse = sqrt(mean_squared_error(test.values, predictions))
print(f'RMSE: {rmse}')