andrei-turcan/snake-deep-reinforcement-learning

The project implements the Snake game as an OpenAI Gym environment. Deep learning is implemented using the RLlib library. A convolutional neural network is used to work with the game frames.

Snake Deep Reinforcement Learning

Description

This is a simple deep reinforcement learning project. The project implements the Snake game as an OpenAI Gym environment. Deep learning is implemented using the RLlib library. A convolutional neural network is used to work with the game frames.

Google Colab

Try it here Colab notebook

Installing dependencies

The project uses Python 3.7.

pip install -r requirements.txt

Train

python train.py

Play

Example:
python play.py checkpoint_example/checkpoint-100