← all repositories

linyiLYi/snake-ai

A deep reinforcement learning project that trains neural network agents to play the classic Snake game.

1.8k stars Python AgentsML Frameworks
snake-ai
Velocity · 7d
+1.6
★ / day
Trend
steady
star history

This project implements AI agents trained via deep reinforcement learning to play the Snake game automatically. It includes two model architectures: a Multi-Layer Perceptron (MLP) agent and a Convolution Neural Network (CNN) agent, with the CNN version achieving higher average scores. Training uses Stable-Baselines3 and PyTorch, with progress tracked via Tensorboard. Pre-trained model weights are provided for testing.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.