Model Card
Browse files- README.md +21 -0
- chess-example.GIF +0 -0
- logo.jpeg +0 -0
README.md
ADDED
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# BertHarmon
|
2 |
+
|
3 |
+
Research done at Johns Hopkins University by Michael DeLeo![iu-13](logo.jpeg)
|
4 |
+
|
5 |
+
Contact: mdeleo2@jh.edu
|
6 |
+
|
7 |
+
## Introduction
|
8 |
+
|
9 |
+
BertHarmon is a BERT model trained for the task of Chess.
|
10 |
+
|
11 |
+
![IMG_0145](chess-example.GIF)
|
12 |
+
|
13 |
+
## Sample Usage
|
14 |
+
|
15 |
+
```python
|
16 |
+
from transformers import pipeline
|
17 |
+
task = pipeline('fill-mask', model='BertHarmon')
|
18 |
+
task("rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1 White <MOVE_SEP> [MASK]")
|
19 |
+
```
|
20 |
+
|
21 |
+
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
|
chess-example.GIF
ADDED
|
logo.jpeg
ADDED
![]() |