search

CSSLab / maia-chess

Maia is a human-like neural network chess engine trained on millions of human games.
https://maiachess.com
GNU General Public License v3.0
963 stars 121 forks source link

How can I train a model personalized to a particular player? #36

Closed FrugoFruit90 closed 2 years ago

FrugoFruit90 commented 2 years ago

Is there a script that would do pretty much the same as train_maia.py, but starting with a given set of weights?

This achieves the highest move-matching accuracy in the paper but I didn't find it mentioned in the readme.

reidmcy commented 2 years ago

This is just the code for the original (KDD 2020) paper. The personalize models paper has not been published yet so we don't have a code release.

bradley-Amuj commented 2 years ago

can I get a personalized model if I use training data that is from one particular player?

reidmcy commented 2 years ago

In theory yes, but we needed about 10 million games to get good performance when starting from scratch. So I don't think there's anyone with enough games for that to work.

alexmolas commented 2 years ago

isn't it possible to use a pre-trained model and then fine-tune it with a small set of games from a particular user?

reidmcy commented 2 years ago

Yes, we're working on the code release for that right now: https://github.com/CSSLab/maia-individual

FrugoFruit90 commented 2 years ago

Would love to use that feature too.