The YGO-Agent framework allows researchers and enthusiasts to develop AI bots that play the Yu-Gi-Oh card game using reinforcement learning. It wraps the YGOPRO game simulator into an OpenAI Gym-compatible environment, defining state representations such as hand, field, and life points, and action representations including summoning, spell/trap activation, and attacking. Rewards are based on win/loss outcomes, damage dealt, and game progress. The agent architecture uses PyTorch to implement DQN, with options for custom network architectures, experience replay, and epsilon-greedy exploration. Logging modules record training curves, win rates, and detailed move logs for analysis. The framework is modular, enabling users to replace or extend components such as the reward function or action space.
