Skip to content

hztBUAA/snake-ai-RL

Repository files navigation

Teach AI To Play Snake! Reinforcement Learning With PyTorch and Pygame

In this Python Reinforcement Learning Tutorial series we teach an AI to play Snake! We build everything from scratch using Pygame and PyTorch. The tutorial consists of 4 parts:

You can find all tutorials on my channel: Playlist

  • Part 1: I'll show you the project and teach you some basics about Reinforcement Learning and Deep Q Learning.
  • Part 2: Learn how to setup the environment and implement the Snake game.
  • Part 3: Implement the agent that controls the game.
  • Part 4: Implement the neural network to predict the moves and train it.

Second Development

  • add the model save and load logic, now you can consistently run the program and use the last result.
  • add the test logic so that you can examine the ability of the trained snake, without using the epsilon greedy policy or updating the paramters of the QNet.

About

An example for learning the RL with a snake.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 3

  •  
  •  
  •