소개

강화학습 알고리즘인 DQN을 사용하여 슈팅 게임을 학습했습니다.

소스코드

출처: https://github.com/5h00T/mini_shooting.git

키보드 입력 정보

상하좌우 화살표: 이동

Z: 메뉴 선택, 공격

1: 학습 자동화 여부

X: 메인 화면으로

Q: 게임 종료

DQN알고리즘

dqn.py를 만들어서 슈팅게임을 학습했습니다. 해당 파일과 Episodes.py의 가중치 파일을 수정하여 레이어의 파라미터 개수를 바꾸거나, load 및 save가 가능합니다.

결과 영상