/RL-Snake-Gym

AI to learn and play snake game using Reinforcement Learning

Primary LanguagePython

RL-Snake-Gym

AI to learn and play snake game using Deep Q Learning.

Installation

git clone https://github.com/PratheepSenthil/RL-Snake-Gym/
cd RL-Snake-Gym
pip install -r requirements.txt

Run

  • Run pre-trained model

      python snake.py 1 10000
    
  • Train own model

      python snake.py 2 10000