Teach AI To Play Snake! Reinforcement Learning With PyTorch and Pygame

In this Python Reinforcement Learning Tutorial series we teach an AI to play Snake! We build everything from scratch using Pygame and PyTorch. The tutorial consists of 4 parts:

You can find all tutorials on my channel: Playlist

• Part 1: I'll show you the project and teach you some basics about Reinforcement Learning and Deep Q Learning.
• Part 2: Learn how to setup the environment and implement the Snake game.
• Part 3: Implement the agent that controls the game.
• Part 4: Implement the neural network to predict the moves and train it.

How to install

# Create virtual environment
python -m venv .venv

# Activate virtual environment
source ./.venv/bin/activate

# Install dependencies
pip install -e .

Evaluating the Neural Network

1. Running the training

python src/snake_ai_pytorch/controllers/train.py

2. Seeing the trained agent play

python src/snake_ai_pytorch/controllers/enjoy.py

To play yourself

python src/snake_ai_pytorch/controllers/play.py