Prophet 라이브러리 사용법

Prophet

시계열 예측 라이브러리

facebook의 데이터 사이언스 팀에서 출시한 오픈 소스 소프트웨어

 

 

예제

Prophet을 파이썬에서 사용하려면

from fbprophet import Prophet

를 선언해야한다.

 

그리고 datetime형식의 컬럼과 예측하고자하는 컬럼을 가져와야 한다.

아래의 데이터프레임을 예로 해보자.

 

그리고 각 컬럼은 ds와 y로 컬럼명을 바꿔줘야 한다.

avocado_prophet_df.columns = ['ds', 'y']
avocado_prophet_df.head()

 

이제 이 데이터로 예측을 해보자.

 

# 1. 변수로 만들고
prophet = Prophet()

# 2. 기존의 데이터로 학습 시키기
prophet.fit(avocado_prophet_df)

# 365일치를 예측하기
# 3. 예측하고자 하는 기간을 정해서, 비어있는 데이터 프레임 만들기
future = prophet.make_future_dataframe(periods = 365)

# 4. 프로펫의 predict 함수에 빈 데이터 프레임을 넣어서, 예측 데이터를 채운다.
forecast = prophet.predict(future)

# yhat == 예측값
forecast

이를가지고 비주얼라이징 하면 아래와 같다.

prophet.plot(forecast)
plt.savefig('chart1.jpg')

prophet.plot_components(forecast)
plt.savefig('chart2.jpg')