제2회 KRX 주식 투자 알고리즘 경진대회

알고리즘 | 정형 | 시계열 | 포트폴리오 구성 | 금융 | 샤프 지수

마감

 

[Jaai] LSTM 현재가 예측

2023.07.30 13:54 2,138 조회 language

안녕하세요. 
우선 재미있는 주제와 흥미로운 코드들로 더 많은 공부를 할 수 있게 도와주신 데이콘 관계자 여러분들과 참가자 여러분들께 진심으로 감사의 인사를 드리고 싶습니다.
사실, 제 코드는 미래를 예측했다기보다는 현재가를 예측했기 때문에 private 점수 등 알고리즘 평가에서는 유효하지 않을 것 같기는 합니다만,
우선 좀 고생한것도 있고 단순히 이렇게 학습시킨 모델도 유효한 값을 냈기 때문에 신기한 마음에 함께 코드를 공유해보도록 하겠습니다.

제 경우, LSTM 모델로 예측값을 구하는 모델링을 시도했습니다. 
(컴퓨터로 파이썬에서 ARIMA를 시도했는데 이상하게 계속 오류가 나서 이것만 올리게 되었네요;; R은 잘되던데;;)
하지만 아직 향후 15일 이후의 값을 예측하는 방법에 대해서는 아직 코드 연습 및 학습이 필요해
가설로 결국 값은 평균값으로 수렴한다를 전제로 모든 모델링 및 예측을 실시했습니다. 

lstm의 경우 그나마 주식시장이 안정화 되어 있던 
20210601~20220531까지 학습을 시켜 모델을 만들었고 
이후, 20210601~20230530까지의 데이터를 넣어 예측값을 구했습니다.

이후, 가설설정에 맞게 실제값(평균값)과의 비교를 통해
수익률 = ((예측값-실제값)/실제값) 으로 계산해 순위를 결정했습니다.
감사합니다.

덧, 변수로 함께 넣은 값은 한국은행, 아시아경제은행, KOSTAT - Korea 기준으로 진행했습니다.

코드