This is a classic air hockey game built using PyGame, where two AI agents compete through Reinforcement Learning powered by PyTorch. Experience the excitement of virtual air hockey, customize hyperparameters like learning rates and rewards, and track AI progress with training analytics. Easy to set up, this repository offers a glimpse of AI-driven air hockey battles. I have built another repo as well where an agent plays Atari Breakout game..Do check that out too 😁😁
-
Clone this repository: https://github.com/CodeGeek04/Air-Hockey-Reinforcement-Learning
-
Navigate to the repository: cd air-hockey-reinforcement-learning
-
Install the required packages: pip install requirements.txt
-
Run the game: python air_hockey.py
-
Wait for a few epochs to see change in performance of agents. Tune hyperparameters according to usage