BertHarmon / README.md
squish's picture
corrected model
67badc6
metadata
thumbnail: https://en.memesrandom.com/wp-content/uploads/2020/11/juega-ajedrez.jpeg
widget:
  - text: >-
      rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1 White <MOVE_SEP>
      [MASK]
  - example_title: Empty Board
  - text: 6Q1/5k2/3P4/1R3p2/P4P2/7Q/6RK/8 b - - 2 60 Black <MOVE_SEP> [MASK]
  - example_title: Late Game Board

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