Snake AI with Pre-trained Model

Pre-trained Snake AI Demo

This Snake AI has been pre-trained to play the game effectively. Watch it navigate and collect food!

• Demo Mode: Watch the pre-trained AI play
• Continue Training: Further improve the AI's performance
• Speed Control: Adjust the game speed for better viewing

The AI uses a Deep Q-Network (DQN) with the following features:

• 12 input features: snake direction, food direction, and danger detection
• Two hidden layers with 256 neurons each
• Pre-trained weights loaded from PyTorch model