Skip to content
Deep Pink is a chess AI that learns to play chess using deep learning.
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
README.md Update README.md May 26, 2016
load.py Allow running play.py without having scikit-learn and h5py installed. Feb 20, 2017
model.pickle New model (trained for 5 days) Sep 17, 2014
parse_game.py Use HDF5 for intermediary format Aug 22, 2014
play.py
reinforcement.py Allow running play.py without having scikit-learn and h5py installed. Feb 20, 2017
train.py Allow running play.py without having scikit-learn and h5py installed. Feb 20, 2017

README.md

deep-pink

Deep Pink is a chess AI that learns to play chess using deep learning. Here is a blog post providing some details about how it works.

There is a pre-trained model in the repo, but if you want to train your own model you need to download pgn files and run parse_game.py. After that, you need to run train.py, preferrably on a GPU machine since it will be 10-100x faster. This might take several days for a big model.

Note that the code is a bit hacky (eg. hardcoded paths in some places) so you might have to modify those to suit your needs.

Dependencies

  • Theano: git clone https://github.com/Theano/Theano; cd Theano; python setup.py install
  • Sunfish: git clone https://github.com/thomasahle/sunfish. You need to add it to PYTHONPATH to be able to play
  • python-chess pip install python-chess
  • scikit-learn (only needed for training)
  • h5py: can be installed using apt-get install python-hdf5 or pip install hdf5 (only needed for training)
You can’t perform that action at this time.