Update train.py

david-carteau · web-flow · commit 54b6bb35277b · 2025-01-14T19:12:33.000+01:00
diff --git a/v1.1/2. training/train.py b/v1.1/2. training/train.py
@@ -247,6 +247,7 @@ def fit(self, X, y, lr=None):
         if CONVERT_CP_TO_WR:
             # inspired by https://www.chessprogramming.org/Pawn_Advantage,_Win_Percentage,_and_Elo
             mt = 1 / (1 + 10**(-mt/400))
+			mt = 2.0 * mt - 1.0
         else:
             # idea: one queen ahead --> mt = +0.9 --> close to / similar to a win ratio of +1.0
             mt /= 1000.0
