Make score output in datagen white relative

bench 4732467
rn5f107s2 · May 18, 2024 · dcbe3c6 · dcbe3c6
1 parent e900748
commit dcbe3c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/Datagen/Datagen.cpp b/src/Datagen/Datagen.cpp
@@ -105,7 +105,7 @@ void playGame(Position &pos, const std::string& filename, u64 &fenCount) {
 
         if (!pos.isCapture(bestMove) && extract<FLAG>(bestMove) != PROMOTION) {
             fens.push(pos.fen());
-            scores.push(score);
+            scores.push(score * (pos.sideToMove ? 1 : -1));
             bestMoves.push(bestMove);
         }
         pos.makeMove(bestMove);