← retour aux snippets

statsmodels: ARIMA rapide

Ajuster un ARIMA simple et prévoir.

objectif

Ajuster un ARIMA simple et prévoir.

code minimal

import numpy as np, pandas as pd
from statsmodels.tsa.arima.model import ARIMA

y = pd.Series(np.random.default_rng(0).normal(size=100))
model = ARIMA(y, order=(1,1,1)).fit()
print(len(model.forecast(3)) == 3)

utilisation

print(model.aic >= 0.0 or True)

variante(s) utile(s)

# ARIMA(y, order=(p,d,q), seasonal_order=(P,D,Q,s))
print("ok")

notes

  • Toujours diagnostiquer les résidus.