머신러닝 예측 프로젝트

개요

WHO에서 공식적으로 선언한 팬데믹 중 20세기 이후 선언된 H1N1과 COVID-19는 1년 이상 지속되며 일상생활뿐만 아니라 사회, 정치, 경제적인 분야까지 많은 영향을 끼쳤다. 특히 COVID-19 팬데믹은 현재까지 진행되고 있다. 본 연구에서는 H1N1과 COVID-19 팬데믹 기간 동안 미국 3대 지수의 변화를 분석하고 향후 1년간의 주가 변동을 예측하여 팬데믹과 주가 사이의 관계를 명확히 분석했다. Graph Attention Network (GAT)를 활용해 팬데믹과 주가 변동 간의 관계를 분석했으며, Exponential Smoothing State Space Model (ETS) 예측모델을 활용해 향후 1년간의 코로나 감염자 수를 예측하고, Prophet 예측 모델을 사용하여 ETS 모델로 예측된 코로나19 감염자 수와 H1N1 팬데믹 기간 동안의 감염자 수를 포함한 변수들을 바탕으로, 향후 1년간 각 지수의 종가 (closing price)를 예측하였다.

결과 및 지표

• Metric A : 0.97 , 0.96, 0.96
• RMSE : 890, 330, 140

기술

Python, GAT, ETS, Prophet.

코드

코드는 아래 링크에서 확인할 수 있습니다.

코드 보기

논문

논문 원본을 다운로드 할 수 있습니다.

PDF 자료

PDF 자료는 아래에서 확인할 수 있습니다.

다운로드