Update chess-gpt-eval/main.py

chess-gpt-eval/main.py

@@ -183,8 +183,8 @@ def calculate_stats(csv_file_path):
         return None
 
     stats = {
-        "wins": sum(float(row["player_one_score"]) for row in data if row["player_one_score"] > 0.6),
-        "draws": len(data) - sum(float(row["player_two_score"]) for row in data if row["player_two_score"] > 0.6) - sum(float(row["player_one_score"]) for row in data if row["player_one_score"] > 0.6),
+        "wins": sum(float(row["player_one_score"]) for row in data if float(row["player_one_score"]) > 0.6),
+        "draws": len(data) - sum(float(row["player_two_score"]) for row in data if float(row["player_two_score"]) > 0.6) - sum(float(row["player_one_score"]) for row in data if float(row["player_one_score"]) > 0.6),
         "illegal_attempts_ratio": sum(float(row["p1_illegal_attempts"]) for row in data) / (sum(float(row["p1_illegal_attempts"]) for row in data) + sum(float(row["player_one_legal_moves"]) for row in data)),
         "illegal_moves_ratio": sum(float(row["player_one_illegal_moves"]) for row in data) / sum(float(row["player_one_illegal_moves"]) + float(row["player_one_legal_moves"]) for row in data),
         "avg_attempts_per_illegal": sum(float(row["p1_avg_attempts_per_illegal"]) for row in data) / len(data),
@@ -633,7 +633,7 @@ def play_games(
 RUN_FOR_ANALYSIS = True
 MAX_MOVES = 999 # Due to nanogpt max input length of 1024
 recording_file = "logs/determine.csv" # default recording file. Because we are using list [player_ones], recording_file is overwritten
-player_ones = ["50M/ckpt_35549900g_12620818850t.pt", "50M/ckpt_36749900g_13053687575t.pt", "50M/ckpt_38849900g_13811553625t.pt", "50M/ckpt_42450000g_15114740300t.pt", "50M/ckpt_49200000g_17555619300t.pt", "50M/ckpt_53549900g_19130809575t.pt", "50M/ckpt_58109500g_20779625975t.pt", "50M/ckpt_61154900g_21867551750t.pt", "50M/ckpt_62759900g_22442258700t.pt", "50M/ckpt_65699900g_23491965075t.pt", "50M/ckpt_67876200g_24270785550t.pt"]
+player_ones = ["50M/ckpt_38849900g_13811553625t.pt", "50M/ckpt_42450000g_15114740300t.pt", "50M/ckpt_49200000g_17555619300t.pt", "50M/ckpt_53549900g_19130809575t.pt", "50M/ckpt_58109500g_20779625975t.pt", "50M/ckpt_61154900g_21867551750t.pt", "50M/ckpt_62759900g_22442258700t.pt", "50M/ckpt_65699900g_23491965075t.pt", "50M/ckpt_67876200g_24270785550t.pt"]
 player_two_recording_name = "lc0_sweep" #"stockfish_sweep"
 move_num_in_gamestate = False
 book_opening = True