ARIMA 모형

import matplotlib.pyplot as plt
import pandas as pd

df = pd.read_excel('beer.xlsx')
df.head()

y = df.production
y.plot()

<Axes: >

from  statsmodels.tsa.api import SARIMAX

arma = SARIMAX(y, order=(1, 0, 1)).fit()
arma.summary() 

y.plot()
arma.predict(0, 250).plot()

<Axes: >

from statsmodels.tsa.stattools import kpss
kpss(y)

C:\Users\eupho\AppData\Local\Temp\ipykernel_28656\4027067367.py:2: InterpolationWarning: The test statistic is outside of the range of p-values available in the
look-up table. The actual p-value is smaller than the p-value returned.

  kpss(y)

(1.1416279223770989,
 0.01,
 9,
 {'10%': 0.347, '5%': 0.463, '2.5%': 0.574, '1%': 0.739})

yd = y.diff(1).dropna()
yd.plot()

<Axes: >

kpss(yd)

C:\Users\eupho\AppData\Local\Temp\ipykernel_28656\757044700.py:1: InterpolationWarning: The test statistic is outside of the range of p-values available in the
look-up table. The actual p-value is greater than the p-value returned.

  kpss(yd)

(0.12919635959269035,
 0.1,
 17,
 {'10%': 0.347, '5%': 0.463, '2.5%': 0.574, '1%': 0.739})

d_arma = SARIMAX(yd, order=(1, 0, 1)).fit()
d_arma.summary() 

c:\Users\eupho\anaconda3\lib\site-packages\statsmodels\tsa\base\tsa_model.py:473: ValueWarning: An unsupported index was provided and will be ignored when e.g. forecasting.
  self._init_dates(dates, freq)
c:\Users\eupho\anaconda3\lib\site-packages\statsmodels\tsa\base\tsa_model.py:473: ValueWarning: An unsupported index was provided and will be ignored when e.g. forecasting.
  self._init_dates(dates, freq)
c:\Users\eupho\anaconda3\lib\site-packages\statsmodels\tsa\statespace\sarimax.py:978: UserWarning: Non-invertible starting MA parameters found. Using zeros as starting parameters.
  warn('Non-invertible starting MA parameters found.'

yd.plot()
d_arma.predict(0, 250).plot()

c:\Users\eupho\anaconda3\lib\site-packages\statsmodels\tsa\base\tsa_model.py:836: ValueWarning: No supported index is available. Prediction results will be given with an integer index beginning at `start`.
  return get_prediction_index(
c:\Users\eupho\anaconda3\lib\site-packages\statsmodels\tsa\base\tsa_model.py:836: FutureWarning: No supported index is available. In the next version, calling this method in a model without a supported index will result in an exception.
  return get_prediction_index(

<Axes: >

arima = SARIMAX(y, order=(1, 1, 1)).fit()
arima.summary() 

c:\Users\eupho\anaconda3\lib\site-packages\statsmodels\tsa\statespace\sarimax.py:978: UserWarning: Non-invertible starting MA parameters found. Using zeros as starting parameters.
  warn('Non-invertible starting MA parameters found.'

y.plot()
arima.predict(0, 250).plot()

<Axes: >

sarima = SARIMAX(y, order=(1, 1, 0), seasonal_order=(1, 0, 0, 4)).fit()
sarima.summary()

y.plot()
sarima.predict(0, 250).plot()

<Axes: >

from itertools import combinations_with_replacement
import numpy as np
import warnings

warnings.filterwarnings("ignore")

min_aic = np.inf
for ar, ma, seasonal_ar, seasonal_ma in combinations_with_replacement([0, 1, 2], 4):
    try:
        sarima = SARIMAX(y, order=(ar, 1, ma), seasonal_order=(seasonal_ar, 0, seasonal_ma, 4)).fit()
        print(ar, ma, seasonal_ar, seasonal_ma, sarima.aic)
    except:
        pass
    else:
        if sarima.aic < min_aic:
            min_aic = sarima.aic
            best_sarima = sarima
warnings.resetwarnings()

0 0 0 0 2446.993870307465
0 0 0 0 2446.993870307465
0 0 0 1 2269.0870570757484
0 0 0 1 2269.0870570757484
0 0 0 2 2205.6411732862007
0 0 0 2 2205.6411732862007
0 0 1 1 1968.7883160730275
0 0 1 1 1968.7883160730275
0 0 1 2 1970.3592365281647
0 0 2 2 1965.3952928674607
0 0 2 2 1965.3952928674607
0 1 1 1 1840.469834804714
0 1 1 1 1840.469834804714
0 1 1 2 1840.7075395781462
0 1 2 2 1836.4792775263727
0 1 2 2 1836.4792775263727
0 2 2 2 1822.26570408957
0 2 2 2 1822.26570408957
1 1 1 1 1827.969616228781
1 1 1 2 1828.8672615487212
1 1 2 2 1824.7291286441794
1 2 2 2 1824.2634784092993

best_sarima.summary()