/snake_rl

Building a reinforcement learning algorithm to solve the snake game

Primary LanguagePython

snake_rl

Building a reinforcement learning algorithm to solve the snake game in response to OpenAI's request for research.

requirements and setup

Install the snake gym:

git clone https://github.com/ChloeL19/c_snake.git c_snake_gym

pip install -e c_snake_gym

Install the DQN code:

git clone https://github.com/ChloeL19/snake_rl.git

Ensure the following are installed:

matplotlib==3.3.3

gym==0.17.3

Keras==2.4.3

tensorflow==2.3.1

instructions for running

Navigate into snake_rl directory.

python DQN.py

Videos of the snake's performance will be saved to the local directory.

Have fun! :)