Skip to content

Latest commit

 

History

History
308 lines (220 loc) · 9.9 KB

README.md

File metadata and controls

308 lines (220 loc) · 9.9 KB

Poker Simulation

Python License GitHub Issues GitHub Stars

Table of Contents

Overview

Poker Simulation is a Python-based project that simulates Texas Hold'em poker games. It leverages reinforcement learning to train agents, evaluate strategies, and analyze gameplay dynamics. The simulation environment is built using the gym library, and the agents utilize neural networks implemented with PyTorch to make strategic decisions.

This project aims to provide a comprehensive framework for studying poker strategies, testing AI agents, and conducting large-scale simulations to understand game dynamics better.

Next‑Gen Upgrade (XGBoost EV + CFR + PPO)

New optional package nextgen/ adds a production‑grade engine, EV prediction and scalable self‑play that match or surpass an R/XGBoost simulator:

  • Mechanics: strict Hold'em betting (min‑raise, side pots), four streets, showdown and tie handling, wheel straights; deterministic hand ranking categories (1–9) with tie‑breakers.
  • Features: ~500 engineered features (ranks, suits, texture, straight/flush draws, position, pot odds, history by player/round). Fixed shape 512 with action encoding.
  • EV model: XGBoost regression for expected value of (state, action) with CV + early stopping; portable fallbacks if XGBoost/SKLearn unavailable.
  • Agents: Random/Human/Smart‑EV; PPO wrapper (stable‑baselines3, optional); simple CFR/regret matching hooks via abstractions.
  • Training: iterative self‑play to generate labeled EV data; nextgen/train_agents.py trains and persists the EV model.
  • Evaluation: nextgen/evaluate.py matches Smart‑EV vs randoms and can plot win rates.

What’s new vs the R code (ai3.R):

  • Parallel self‑play via Ray (36 workers recommended), designed to scale to 2M+ hands with chunking and seed control.
  • Optuna hyperparameter tuning for the XGBoost EV model.
  • Approximate exploitability metric using a CFR/regret‑matching toy preflop node for quick sanity tracking.
  • Deterministic, tie‑breaking hand evaluator (A‑low wheel, straight flushes, side pots) with precise rank vectors.

Quick start:

pip install -r requirements.txt  # see added deps
python -m nextgen.sample_run     # 5 demo hands and payouts
python -m nextgen.train_agents --games 5000 --out artifacts
python -m nextgen.evaluate --model artifacts --games 200 --plot

VS Code: open nextgen/ files, press F5 to debug a module (Python); see .vscode/launch.json examples.

Advanced nextgen usage:

  • Parallel dataset: python -m nextgen.train_agents --games 200000 --parallel --workers 36 --out artifacts
  • With Optuna tuning (xgboost backend): python -m nextgen.train_agents --games 500000 --parallel --workers 36 --tune 25
  • Evaluate + toy exploitability metric: python -m nextgen.evaluate --model artifacts --games 1000 --exploit --plot

Sample output (python -m nextgen.sample_run):

Hand 1 community=[6H, KC, 3C, TS, AH] payouts={0: 0, 1: 0, 2: 151, 3: 0} pot=151
Hand 2 community=[5H, JD, 2S, AC, TS] payouts={0: 0, 1: 2, 2: 0, 3: 0} pot=2
Hand 3 community=[6C, 5H, KC, QD, TC] payouts={0: 0, 1: 5, 2: 0, 3: 0} pot=5
...

Features

  • Modular Architecture: Organized into distinct modules for classes, utilities, simulation, and environment management.
  • Reinforcement Learning Agents: Agents trained using neural networks to make informed betting decisions.
  • Comprehensive Evaluation: Tools to evaluate agent performance, select top-performing players, and analyze average pot winnings.
  • Extensible Framework: Easily add new strategies, agents, or modify game rules.
  • Logging and Reporting: Detailed logs and reports to track simulation progress and outcomes.
  • Unit Testing: Ensures code reliability and correctness through extensive unit tests.

Installation

Prerequisites

  • Python 3.8 or higher: Ensure you have Python installed. You can download it from here.
  • Git: To clone the repository. Download from here.

Steps

  1. Clone the Repository

    git clone https://github.com/yourusername/poker_simulation.git
cd poker_simulation

  2. Create a Virtual Environment

    It's recommended to use a virtual environment to manage dependencies.

    python -m venv venv

  3. Activate the Virtual Environment

    On Unix/Linux/MacOS:

    source venv/bin/activate

    On Windows:

    venv\Scripts\activate

  4. Install Dependencies

    Install the required dependencies from requirements.txt:

    pip install -r requirements.txt

    If requirements.txt is not present, you can manually install the necessary packages:

    pip install gym torch numpy pandas treys tqdm

Usage

Running Simulations

To run a simulation of poker games, execute the main.py script. You can specify parameters such as the number of generations, games per generation, and more.

python main.py --simulate

Example with custom parameters:

python main.py --simulate --generations 50 --games 1000 --chunk_size 100

Training Agents

To train poker agents using reinforcement learning, utilize the training scripts provided in the simulation/ directory.

python simulation/train_agents.py

Evaluating Agents

Evaluate the performance of trained agents using the evaluation tools.

python utils/evaluation.py

Note: Replace with actual script paths and commands based on your implementation.

ROCm (AMD GPUs)

  • Tested on Radeon RX 7900 XT (20 GB) with PyTorch ROCm.
  • Use the GPU-optimized runner to avoid disabling the GPU and to apply ROCm allocator tuning:
chmod +x run_train_rocm.sh
./run_train_rocm.sh --any-train-flags
  • The script sets HIP_VISIBLE_DEVICES=0 and PYTORCH_HIP_ALLOC_CONF=garbage_collection_threshold:0.6,expandable_segments:True,max_split_size_mb:256.
  • If you previously used run_train.sh, note it exports POKERAI_DISABLE_GPU=1 (CPU‑only). Prefer run_train_rocm.sh for GPU runs.
  • Mixed precision is enabled by default. The code probes bfloat16/float16 support and falls back safely.
  • VRAM checks use reserved memory on ROCm; training auto‑reduces batch sizes upon OOM.

Project Structure

poker_simulation/
│
├── classes/
│   ├── __init__.py
│   ├── player.py
│   ├── card.py
│   └── game_state.py
│
├── environment/
│   ├── __init__.py
│   └── poker_env.py
│
├── simulation/
│   ├── __init__.py
│   ├── simulate.py
│   └── train_agents.py
│
├── utils/
│   ├── __init__.py
│   ├── helper_functions.py
│   ├── training.py
│   └── evaluation.py
│
├── tests/
│   ├── __init__.py
│   ├── test_game_state.py
│   └── test_evaluation.py
│
├── logs/
│   └── simulation.log
│
├── main.py
├── requirements.txt
├── README.md
└── LICENSE

Description

  • classes/: Contains core classes such as Player, Card, and GameState that define the fundamental components of the poker game.
  • environment/: Defines the simulation environment using gym, encapsulating the game logic and state management.
  • simulation/: Houses scripts related to running simulations, training agents, and generating game data.
  • utils/: Utility modules for helper functions, training utilities, and evaluation tools.
  • tests/: Unit tests to ensure code reliability and correctness.
  • logs/: Directory for storing log files generated during simulations.
  • main.py: The entry point for running simulations and managing the workflow.
  • requirements.txt: Lists all Python dependencies required for the project.
  • README.md: Documentation of the project.
  • LICENSE: Licensing information.

Running Tests

Unit tests are provided to validate the functionality of various components.

To run all tests, execute:

python -m unittest discover tests

Alternatively, use pytest if preferred:

Install pytest

pip install pytest

Run Tests

pytest

Contributing

Contributions are welcome! Follow these steps to contribute:

Fork the Repository

Create a New Branch

git checkout -b feature/YourFeatureName

Make Your Changes

Commit Your Changes

git commit -m "Add some feature"

Push to the Branch

git push origin feature/YourFeatureName

Open a Pull Request

Please ensure your code adheres to the project's coding standards and passes all tests.

License

This project is licensed under the MIT License. You are free to use, modify, and distribute it as per the license terms.

Contact

Aaron Esbenshade
Email: ajesbenshade@outlook.com
GitHub: ajesbenshade
LinkedIn: Your LinkedIn

Feel free to reach out for questions, suggestions, or collaboration opportunities.

Acknowledgments

  • Gym for providing the reinforcement learning environment framework.
  • PyTorch for the deep learning library.
  • Treys for poker hand evaluation.
  • TQDM for progress bars in simulations.
  • OpenAI for their contributions to AI and machine learning.