aabha-j / snake-game-bot

Play the classic snake game or train a bot using neural networks in pytorch to play it for you

Overview 🐍

This project implements a classic Snake game using Pygame and trains an AI agent with pytorch using Deep Q Learning to play the game. The AI learns to navigate the grid, collect food, and avoid collisions with its own body and the walls.

Set Up ⚙️

1. Clone this repo
2. Create and activate a python virtual enviorment
3. Install dependencies pip install pytorch numpy pygame

Play the Game Yourself 🎮

In the terminal: python snake_game.py

Watch the AI Train & Play 🤖

In the terminal: python agent.py