File size: 3,510 Bytes
90d0815 c34b3dc 90d0815 71f4997 90d0815 71f4997 90d0815 71f4997 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 |
---
license: cc-by-nc-4.0
library_name: onnx
tags:
- chess
- deep-learning
- pytorch
- onnx
- strategy
- game-ai
- basic model
datasets:
- Rafs-an09002/chessmate-opening-stats
language:
- en
pipeline_tag: reinforcement-learning
---
# ♟️ ChessMate AI - CNN Evaluation Model
<div align="center">
[](https://creativecommons.org/licenses/by-nc/4.0/)
[**Live Demo**](https://chessmate-engine.onrender.com) • [**GitHub Repository**](https://github.com/Rafsan1711/Chessmate-Engine)
</div>
## 📖 Model Description
This is a **Convolutional Neural Network (CNN)** trained to evaluate chess positions. It takes a board state as input and outputs a scalar evaluation score between `-1` (Black winning) and `+1` (White winning).
It is the core "brain" of the **ChessMate AI** project, designed to run efficiently in web browsers using `onnxruntime-web`.
- **Architecture:** 3-Layer CNN with Batch Normalization and ReLU activation.
- **Framework:** Trained in PyTorch, exported to ONNX (Opset 14).
- **Size:** ~7.5 MB (Highly optimized for web loading).
- **Training Data:** 100,000+ Master-level games from Lichess (Standard Rated > 2000 ELO).
## 🛠️ Technical Specifications
### Input Shape
The model expects a Tensor of shape `(1, 12, 8, 8)` representing the board state using One-Hot Encoding.
- **Channels (12):**
- 0-5: White Pieces (Pawn, Knight, Bishop, Rook, Queen, King)
- 6-11: Black Pieces (Pawn, Knight, Bishop, Rook, Queen, King)
- **Dimensions (8x8):** The chess board squares.
### Output
- **Shape:** `(1, 1)`
- **Value:** Float between `-1.0` and `1.0`.
- `> 0`: Advantage White
- `< 0`: Advantage Black
- `~ 0`: Equal/Draw
## 💻 Usage (JavaScript / ONNX.js)
This model is designed to be used directly in the browser via `onnxruntime-web`.
```javascript
import * as ort from 'onnxruntime-web';
// 1. Load the session
const session = await ort.InferenceSession.create('./chess_model.onnx');
// 2. Prepare Input (Float32Array of size 12*8*8)
// Convert FEN string to 12x8x8 one-hot encoded array
const inputData = new Float32Array(768).fill(0);
// ... (Fill array based on piece positions) ...
const tensor = new ort.Tensor('float32', inputData, [1, 12, 8, 8]);
// 3. Run Inference
const results = await session.run({ board_state: tensor });
const evaluation = results.evaluation.data[0];
console.log(`Position Score: ${evaluation}`);
```
## 🧠 Training Details
- **Loss Function:** Mean Squared Error (MSE)
- **Optimizer:** Adam (lr=0.001)
- **Epochs:** 50 (with Early Stopping)
- **Target Label:** Normalized Stockfish Evaluation / Game Result (Win/Loss/Draw).
## ⚠️ License & Limitations
This model is licensed under **CC BY-NC 4.0** (Attribution-NonCommercial 4.0 International).
**You are free to:**
- Use this model for research, education, and personal projects.
- Modify and adapt the model.
**You may NOT:**
- Sell this model or use it in a commercial product without permission.
---
<div align="center">
<p>Curated by <a href="https://github.com/GambitFlow">GambitFlow</a></p>
</div>
4 |