snakeAI: Reinforcement Learning Agent for Snake

This project features an AI trained to play the classic Snake game using Reinforcement Learning. The implementation leverages PyTorch for the neural network and Pygame for game visualization.

Requirements

To set up the environment, ensure the following dependencies are installed:

• Python 3.8 or higher
• PyTorch
• Pygame
• NumPy
• Matplotlib (optional, for training visualizations)

Usage

Step 1: Clone the Repository

git clone https://github.com/Kongesque/snakeAI
cd snakeAI

Step 2: Train the AI

python agent.py