srmaarnav / snake-ai

Python Reinforcement Learning, teaching AI to play Snake

Snake-AI

Python Reinforcement Learning, teaching AI to play Snake

Table of contents

• General info
• Technologies
• Instructions
• Illustrations

General info

A reinforcement learning based snake player, that tries to play the classic snake game. I used Pygame library for the game, matplotlib for the plotting of the game score and Q-function to get the best states in a reinfocement learning environment.

Technologies

• Python 3.11.1
• pygame 2.3.0
• torch 2.0.0

Instructions

1. Clone the repository and install all the dependecies.
2. Run the python file named main.py.
3. Enjoy the snake trying to improve itself.

Illustrations