alpha-zero-general

(★ 4,502)

A clean implementation based on AlphaZero for any game in any framework + tutorial + Othello/Gobang/TicTacToe/Connect4 and more

File Explorer

Download Latest Version (.zip)

- - __init__.py
  - Connect4NNet.py
  - NNet.py
- __init__.py
- Connect4Game.py
- Connect4Logic.py
- Connect4Players.py
- README.md
- test_connect4.py
- Dockerfile.pytorch
- jupyter_notebook_config.py
- requirements.txt
- run_jupyter.sh
- - __init__.py
  - DotsAndBoxesNNet.py
  - NNet.py
- __init__.py
- Convert Keras Model for use with Tensorflow.js.ipynb
- DotsAndBoxesGame.py
- DotsAndBoxesLogic.py
- DotsAndBoxesPlayers.py
- DotsAndBoxesServer.py
- DotsAndBoxesTest.py
- main-dotsandboxes.py
- pit-dotsandboxes.py
- Play Dots and Boxes using AlphaZero General.ipynb
- README.md
- Train Dots and Boxes using AlphaZero General.ipynb
- - GobangNNet.py
  - NNet.py
- __init__.py
- GobangGame.py
- GobangLogic.py
- GobangPlayers.py
- - __init__.py
  - NNet.py
  - OthelloNNet.py
- - __init__.py
  - NNet.py
  - OthelloNNet.py
- __init__.py
- OthelloGame.py
- OthelloLogic.py
- OthelloPlayers.py
- README.md
- 6x6.png
- writeup.pdf
- - NNet.py
  - RTSNNet.py
- - README.md
- - Board.py
  - config.py
  - config_class.py
  - encoders.py
- - rts_pygame.py
  - rts_ue4.py
- install.sh
- learn.py
- pit.py
- README.md
- RTSGame.py
- RTSPlayers.py
- SANTORINI_README.md
- SantoriniGame.py
- SantoriniLogic.py
- SantoriniPlayers.py
- - __init__.py
  - NNet.py
  - TaflNNet.py
- - __init__.py
  - NNet.py
  - TaflNNet.py
- __init__.py
- Digits.py
- GameVariants.py
- mainTafl.py
- pitTafl.py
- TaflGame.py
- TaflLogic.py
- TaflPlayers.py
- - __init__.py
  - NNet.py
  - TicTacToeNNet.py
- - best.pth.tar.data-00000-of-00001
- __init__.py
- README.md
- TicTacToeGame.py
- TicTacToeLogic.py
- TicTacToePlayers.py
- - __init__.py
  - NNet.py
  - TicTacToeNNet.py
- README.md
- TicTacToeGame.py
- TicTacToeLogic.py
- TicTacToePlayers.py
.gitignore
Arena.py
Coach.py
Game.py
LICENSE
main.py
MCTS.py
NeuralNet.py
pit.py
README.md
requirements.txt
setup_env.sh
test_all_games.py
utils.py

🚀 Installation Guide

1. Get the code

git clone https://github.com/suragnair/alpha-zero-general

Downloads the entire project code from GitHub to your computer.

cd alpha-zero-general

Moves into the project folder you just downloaded.

2. Docker

Easy Recommended

Prerequisites

Git Needed to download the project code from GitHub.
Docker Desktop Needed to build and run containers. Install it and keep it running in the background.

docker exec -ti pytorch_notebook python main.py

Runs the Python script (or module).

✅ Run docker compose ps to check the containers are Up. If the README mentions a port, open http://localhost:PORT in your browser.

Pulled directly from this repo's README.

3. Python

Easy

Prerequisites

Git Needed to download the project code from GitHub.
Python 3 On Windows, be sure to check 'Add Python to PATH' during installation.

pip install -r docker/requirements.txt

Installs the Python libraries listed in requirements.txt (or similar).

python main.py

Runs the Python script (or module).

✅ If it runs without errors and prints output in the terminal, it worked.