Update README.md
Browse files
README.md
CHANGED
|
@@ -6,6 +6,25 @@ tags:
|
|
| 6 |
- game-ai
|
| 7 |
- pytorch
|
| 8 |
library_name: transformers
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
---
|
| 10 |
|
| 11 |
# ChessBot Chess Model
|
|
@@ -61,6 +80,8 @@ print(f"Predicted move: {move}")
|
|
| 61 |
- **Feed-forward size**: 736
|
| 62 |
- **Attention heads**: 8
|
| 63 |
- **Vocabulary size**: 1929 (chess moves)
|
|
|
|
|
|
|
| 64 |
|
| 65 |
## Training Data
|
| 66 |
|
|
|
|
| 6 |
- game-ai
|
| 7 |
- pytorch
|
| 8 |
library_name: transformers
|
| 9 |
+
model-index:
|
| 10 |
+
- name: ChessBot
|
| 11 |
+
results:
|
| 12 |
+
- task:
|
| 13 |
+
type: other
|
| 14 |
+
name: Chess Move Prediction
|
| 15 |
+
metrics:
|
| 16 |
+
- type: accuracy
|
| 17 |
+
name: Legal Move Accuracy
|
| 18 |
+
value: 0.95
|
| 19 |
+
- type: other
|
| 20 |
+
name: Parameters
|
| 21 |
+
value: 31.7M
|
| 22 |
+
base_model_revision: main
|
| 23 |
+
widget:
|
| 24 |
+
- text: "rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKQ w KQkq - 0 1"
|
| 25 |
+
example_title: "Starting Position"
|
| 26 |
+
- text: "r1bqkb1r/pppp1ppp/2n2n2/1B2p3/4P3/5N2/PPPP1PPP/RNBQK2R w KQkq - 4 4"
|
| 27 |
+
example_title: "Italian Game"
|
| 28 |
---
|
| 29 |
|
| 30 |
# ChessBot Chess Model
|
|
|
|
| 80 |
- **Feed-forward size**: 736
|
| 81 |
- **Attention heads**: 8
|
| 82 |
- **Vocabulary size**: 1929 (chess moves)
|
| 83 |
+
- **Total parameters**: 31,708,102 (~31.7M)
|
| 84 |
+
- **Model size**: ~127MB (safetensors)
|
| 85 |
|
| 86 |
## Training Data
|
| 87 |
|