강화학습을 이용한 주식 트레이더 AI

• Mac

• Python 3

• Tensorflow 2.5
• Keras 2.5
• Numpy 1.19.5
• Pandas 1.3.2
• Matplotlib 3.4.3

• 투자 행동 수행
• 투자금과 보유 주식 관리

• 차트 데이터와 학습 데이터 생성
• 자질 벡터 정의
• 데이터 전처리

• 투자할 종목의 차트 데이터를 관리

• 다양한 강화학습 방식을 수행

• 다양한 조건으로 강화학습을 수행할 수 있게 프로그램 인자를 구성
• 입력받은 인자에 따라 학습기 클래스를 이용해 강화학습을 수행
• 학습한 신경망들을 저장

• 가치 신경망과 정책 신경망을 사용하기 위한 신경망(DNN, LSTM, CNN)

• 경로 설정
• 로케일 설정

• 시간 문자열 생성 함수
• 시그모이드(sigmoid) 함수

• 신경망을 학습하는 과정에서 보유 주식수, 가치 신경망 출력, 정책 신경망 출력, 정책 신경망 출력, 투자 행동, 포트폴리오 가치 등을 시간에 따라 연속으로 보여주기 위한 시각화 기능을 담당

파이썬과 케라스를 이용한 딥러닝/강화학습 주식투자 (개정판)을 참고했습니다.