예제에서 ARIMA(5,1,0)을 적용해 보겠습니다. 자기 회귀 차수를 5로 설정하고 차분 차수는 1을 사용하겠습니다. 또한, 시계열을 정지 상태로 만들기 위해 이동 평균 차수는 0을 사용합니다.

예제는 두 단계로 진행됩니다. 첫 번째 단계에서 ARIMA() 함수를 사용하여 간단한 오차 정보만 보여 주는 예제를 먼저 진행한 후, 두 번째 단계에서 첫 번째 단계를 확장하여 실제 예측해 봅니다(코드를 좀 더 단순화하여 이해하기 쉽게 하기 위해 두 단계로 나누어 진행하는 것이며, 특별한 의미는 없습니다).

먼저 statsmodels 라이브러리를 설치합니다.

> conda install -c conda-forge statsmodels

혹은

> pip install statsmodels

설치가 완료되었다면, 첫 번째 단계의 예제를 구현하는 데 필요한 라이브러리와 데이터를 호출합니다. 데이터셋은 7장 예제 파일의 data 폴더에 있는 sales.csv 파일²을 사용하며, 자전거 매출 정보가 담겨 있습니다.