Enh(analysis): Minor plotting tweaks

Author: klieret

codeclash/analysis/viz/heatmap_win_rates.py

@@ -13,7 +13,7 @@
 from codeclash.constants import LOCAL_LOG_DIR
 
 
-def main(log_dir: Path, unit: str = "rounds", output_file: Path = ASSETS_DIR / "heatmap_win_rates.png"):
+def main(log_dir: Path, unit: str = "rounds", output_file: Path = ASSETS_DIR / "heatmap_win_rates.pdf"):
     print(f"Creating win rate heatmap from logs in {log_dir}...")
 
     # Track round-level wins: (model1, model2) -> [wins, total_rounds]
@@ -71,35 +71,37 @@ def main(log_dir: Path, unit: str = "rounds", output_file: Path = ASSETS_DIR / "
     # Plot
     FONT_BOLD.set_size(18)
     _, ax = plt.subplots(figsize=(10, 8))
-    cmap = mcolors.LinearSegmentedColormap.from_list("br", ["#e74c3c", "#ffffff", "#3498db"])
+    cmap = mcolors.LinearSegmentedColormap.from_list("rg", ["#f44336", "#ffffff", "#4caf50"])  # Red-Green
+
     masked = np.ma.masked_where(np.isnan(matrix), matrix)
     ax.imshow(masked, cmap=cmap, vmin=0, vmax=1)
 
     # Add percentages
     for i in range(n):
         for j in range(n):
             if not np.isnan(matrix[i, j]):
-                color = "white" if abs(matrix[i, j] - 0.5) > 0.3 else "black"
                 ax.text(
                     j,
                     i,
                     f"{matrix[i, j]:.0%}",
                     ha="center",
                     va="center",
-                    color=color,
+                    color="black",
                     fontweight="bold",
                     fontproperties=FONT_BOLD,
+                    fontsize=FONT_BOLD.get_size() + 2,
                 )
 
     ax.set_xticks(range(n))
     ax.set_yticks(range(n))
     ax.set_xticklabels(clean_names, rotation=45, ha="right", fontproperties=FONT_BOLD)
     ax.set_yticklabels(clean_names, fontproperties=FONT_BOLD)
+    ax.tick_params(length=0)
 
     # plt.colorbar(im, label="Win Rate")
     plt.tight_layout()
     if unit == "tournaments":
-        suffix = output_file.suffix or ".png"
+        suffix = output_file.suffix or ".pdf"
         output_file = output_file.with_name(f"{output_file.stem}_tournaments{suffix}")
     plt.savefig(output_file, dpi=300, bbox_inches="tight")
     print(f"Heatmap saved to {output_file}")
@@ -110,7 +112,7 @@ def main(log_dir: Path, unit: str = "rounds", output_file: Path = ASSETS_DIR / "
     parser.add_argument("-d", "--log_dir", type=Path, default=LOCAL_LOG_DIR, help="Path to logs")
     parser.add_argument("-u", "--unit", type=str, default="rounds", help="Unit of analysis: rounds or tournaments")
     parser.add_argument(
-        "-o", "--output_file", type=Path, default=ASSETS_DIR / "heatmap_win_rates.png", help="Output file path"
+        "-o", "--output_file", type=Path, default=ASSETS_DIR / "heatmap_win_rates.pdf", help="Output file path"
     )
     args = parser.parse_args()
     main(**vars(args))

codeclash/analysis/viz/line_chart_per_round_win_rate.py

@@ -100,7 +100,8 @@ def aggregate_win_rates_across_games(profiles):
     # Create line chart of win rate progression per player
     plt.figure(figsize=(6, 6))
     idx = 0
-    for pid, win_rates in lines.items():
+    # Sort lines by model name to get consistent legend order
+    for pid, win_rates in sorted(lines.items()):
         plt.plot(
             range(1, 16),
             win_rates,
@@ -122,28 +123,22 @@ def aggregate_win_rates_across_games(profiles):
         fontproperties=FONT_BOLD,
         fontsize=14,
     )
+    plt.gca().set_yticks([i / 20 for i in range(2, 21)], minor=True)
     plt.ylim(0.1, 1)
     FONT_BOLD.set_size(14)
-    # Get handles and labels, then sort by label alphabetically
-    handles, labels = plt.gca().get_legend_handles_labels()
-    sorted_pairs = sorted(zip(handles, labels), key=lambda x: x[1])
-    sorted_handles, sorted_labels = zip(*sorted_pairs)
-
     plt.legend(
-        sorted_handles,
-        sorted_labels,
         bbox_to_anchor=(1, 1),
         loc="upper right",
         ncol=2,
         prop=FONT_BOLD,
         handletextpad=0.3,
         borderpad=0.3,
-        handlelength=0.5,
+        handlelength=1.5,
     )
     plt.grid(True, alpha=0.3)
     plt.tight_layout()
-    plt.savefig(ASSETS_DIR / "line_chart_per_round_win_rate.png", dpi=300, bbox_inches="tight")
-    print("Win rate progression chart saved to line_chart_per_round_win_rate.png")
+    plt.savefig(ASSETS_DIR / "line_chart_per_round_win_rate.pdf", dpi=300, bbox_inches="tight")
+    print("Win rate progression chart saved to line_chart_per_round_win_rate.pdf")
 
     # Print summary statistics
     print("\n" + "=" * 50)

codeclash/analysis/viz/recover_after_loss_streak.py

@@ -201,7 +201,6 @@ def plot_comeback_probabilities(
     """Plot probability of winning after i consecutive losses."""
     fig, ax = plt.subplots(figsize=(6, 6))
     label_font = FontProperties(fname=FONT_BOLD.get_file(), size=18)
-    title_font = FontProperties(fname=FONT_BOLD.get_file(), size=18)
     legend_font = FontProperties(fname=FONT_BOLD.get_file(), size=14)
 
     sorted_models = sorted(comeback_prob_df.index)
@@ -239,7 +238,6 @@ def plot_comeback_probabilities(
 
     ax.set_xlabel("Number of Consecutive Rounds Lost", fontproperties=label_font)
     ax.set_ylabel("Probability of winning next round", fontproperties=label_font)
-    ax.set_title("Comeback probability after loss streak", fontproperties=title_font)
     ax.legend(loc="upper right", prop=legend_font)
     ax.grid(True, alpha=0.3)
     ax.yaxis.set_minor_locator(AutoMinorLocator())
@@ -261,7 +259,6 @@ def plot_falldown_probabilities(
     """Plot probability of losing after i consecutive wins."""
     fig, ax = plt.subplots(figsize=(6, 6))
     label_font = FontProperties(fname=FONT_BOLD.get_file(), size=18)
-    title_font = FontProperties(fname=FONT_BOLD.get_file(), size=18)
     legend_font = FontProperties(fname=FONT_BOLD.get_file(), size=14)
 
     sorted_models = sorted(falldown_prob_df.index)
@@ -299,7 +296,6 @@ def plot_falldown_probabilities(
 
     ax.set_xlabel("Number of Consecutive Rounds Won", fontproperties=label_font)
     ax.set_ylabel("Probability of losing next round", fontproperties=label_font)
-    ax.set_title("Falldown probability after win streak", fontproperties=title_font)
     ax.legend(loc="upper right", prop=legend_font)
     ax.grid(True, alpha=0.3)
     ax.yaxis.set_minor_locator(AutoMinorLocator())