다음은 statsmodels 라이브러리를 이용한 sales 데이터셋에 대한 예측을 실행한 결과입니다.
predicted=354.377730, expected=346.300000
predicted=288.627290, expected=329.700000
predicted=382.817953, expected=445.400000
predicted=339.543839, expected=325.900000
predicted=392.897253, expected=449.300000
predicted=354.488010, expected=411.300000
predicted=452.200100, expected=417.400000
predicted=406.806117, expected=545.500000
predicted=430.162052, expected=477.600000
predicted=492.745314, expected=687.000000
predicted=493.604679, expected=435.300000
predicted=657.397158, expected=587.300000
predicted=522.091111, expected=676.900000
Test MSE: 8074.991
▲ 그림 7-3 예제에 대한 예측 결과
실제 데이터(빨간색)와 모형 실행 결과(파란색)를 표시한 그림이 만들어졌습니다. 데이터가 우상향 추세를 나타내고 있으므로, 자전거 판매가 향후에도 계속 증가할 것임을 예측할 수 있습니다. 이와 같이 ARIMA를 사용할 경우 데이터 경향을 파악해서 미래를 예측할 수 있습니다.