Python으로 배우는 ARIMA 모델
James Fulton
Climate informatics researcher
# 모델 생성 model = ARIMA(df, order=(1,0,1))# 모델 적합 results = model.fit()# 요약 출력 print(results.summary())
Statespace Model Results
==============================================================================
Dep. Variable: y No. Observations: 1000
Model: SARIMAX(2, 0, 0) Log Likelihood -1399.704
Date: Fri, 10 May 2019 AIC 2805.407
Time: 01:06:11 BIC 2820.131
Sample: 01-01-2013 HQIC 2811.003
- 09-27-2015
Covariance Type: opg
# 모델 생성 model = ARIMA(df, order=(1,0,1))# 모델 적합 results = model.fit()# AIC와 BIC 출력 print('AIC:', results.aic) print('BIC:', results.bic)
AIC: 2806.36
BIC: 2821.09
# AR 차수 반복 for p in range(3): # MA 차수 반복 for q in range(3):# 모델 적합 model = ARIMA(df, order=(p,0,q)) results = model.fit()# 모델 차수와 AIC/BIC 출력 print(p, q, results.aic, results.bic)
0 0 2900.13 2905.04
0 1 2828.70 2838.52
0 2 2806.69 2821.42
1 0 2810.25 2820.06
1 1 2806.37 2821.09
1 2 2807.52 2827.15
...
order_aic_bic =[] # AR 차수 반복 for p in range(3): # MA 차수 반복 for q in range(3):# 모델 적합 model = ARIMA(df, order=(p,0,q)) results = model.fit()# 차수와 점수를 리스트에 추가 order_aic_bic.append((p, q, results.aic, results.bic))
# 모델 차수와 AIC/BIC로 DataFrame 생성
order_df = pd.DataFrame(order_aic_bic, columns=['p','q', 'aic', 'bic'])
# AIC로 정렬
print(order_df.sort_values('aic'))
p q aic bic
7 2 1 2804.54 2824.17
6 2 0 2805.41 2820.13
4 1 1 2806.37 2821.09
2 0 2 2806.69 2821.42
...
# BIC로 정렬
print(order_df.sort_values('bic'))
p q aic bic
3 1 0 2810.25 2820.06
6 2 0 2805.41 2820.13
4 1 1 2806.37 2821.09
2 0 2 2806.69 2821.42
...
# 모델 적합
model = ARIMA(df, order=(2,0,1))
results = model.fit()
ValueError: Non-stationary starting autoregressive parameters
found with `enforce_stationarity` set to True.
# AR 차수 반복 for p in range(3): # MA 차수 반복 for q in range(3):# 모델 적합 model = ARIMA(df, order=(p,0,q)) results = model.fit() # 모델 차수와 AIC/BIC 출력 print(p, q, results.aic, results.bic)
# AR 차수 반복 for p in range(3): # MA 차수 반복 for q in range(3):try: # 모델 적합 model = ARIMA(df, order=(p,0,q)) results = model.fit() # 모델 차수와 AIC/BIC 출력 print(p, q, results.aic, results.bic)except: # 실패 시 AIC/BIC를 None으로 출력 print(p, q, None, None)
Python으로 배우는 ARIMA 모델