KeshavAnandCode
3eccd97536
- Implement piece-square feature extraction - 32 active features for 32 pieces on board - Tests for feature extraction (7 tests) - Fix: piece_sq * 6 + piece_type mapping
Chess NNUE Distillation
Train a single linear layer on Stockfish's NNUE features.
Quick Start
cd python
source .venv/bin/activate
pip install torch --index-url https://download.pytorch.org/whl/cu121
pip install numpy python-chess tqdm matplotlib h5py joblib pytest
python train_full.py
Architecture
- Input: 61,072 features (352 HalfKAv2_hm + 60,720 FullThreats)
- Output: 1 scalar (centipawns)
- Optimizer: Adam (lr=1e-3, wd=1e-4)