신한AI, 보다 나은 금융 생활을 위한 AI 서비스 아이디어 경진대회

주린이들에겐 너무 어려운 주식 투자!!
주식 투자를 위해 참고해야 할 정보를 집약적이고 이해하기 쉽게 제공할 수는 없을까??

안녕하세요! 저희는 주린이주목!팀입니다.
저희 팀은 주식 초보자들도 쉽게 참고할 수 있는 투자 인사이트를 제공하기 위한, 주식 매수/매도 추천 서비스를 제작해보았습니다.
다양한 경제지표, 주식 거래 데이터 뿐만 아니라 여론을 반영하기 위한 기사, 검색량, 증권 토론방 데이터를 활용한 분류 모델을 개발했습니다.
앞으로 5일 동안 등락률이 얼마나 상승/하락할지를 기반으로, 오늘 매수할지, 유지할지, 매도할지에 대해 추천해주는 모델입니다.
여러 종목에 적용 가능한 강건한 모델 설계를 목표로 3가지 주식 종목에 대해 모델을 적합하고 성능을 확인해보았습니다.

클래스 불균형 문제로 인해 대부분의 모델이 '유지'인 경우만 잘 맞추는 문제를 해결하는 데 집중하여 모델 고도화를 진행했습니다.
전체 정확도보다는 가장 중요한 '매수', '매도'의 정확도를 높이면서도 '유지'의 정확도도 유지하는 모델을 개발하고자 했습니다.
수많은 모델링 시도 끝에, XGBoost Classifier와 LSTM Regressor를 활용한 두 가지 모델을 제안합니다.

4월 초부터 준비 많이 한 프로젝트입니다. 좋게 봐주시면 감사드리겠습니다!
읽어주셔서 감사합니다.