AI_That_Plays_Snake

Optimized_models - contains best model which is trained

PlayAgent.py - Is the driving agent

model.py - is the pytorch model

Snakegame_AI.py - contains snake game model built using pygame