Edit model card

BertHarmon

Research done at Johns Hopkins University by Michael DeLeo

Contact: mdeleo2@jh.edu

iu-13

Introduction

BertHarmon is a BERT model trained for the task of Chess.

IMG_0145

Sample Usage

from transformers import pipeline
task = pipeline('fill-mask', model='squish/BertHarmon')
task("rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1 White <MOVE_SEP> [MASK]")

The base string consists of the FEN_position followed by the player color and a move seperator. Finally with the [MASK] token. The mask token is the algebraic notation for a chess move to be taken givent the current board state in FEN Notation

Links

Github

HuggingFace

Downloads last month
2