wip

tools/plot_scripts/results_latent_space_comparisons.py

@@ -7,10 +7,10 @@ from pathlib import Path
 import matplotlib.pyplot as plt
 import numpy as np
 import polars as pl
-from matplotlib.lines import Line2D
 
 # CHANGE THIS IMPORT IF YOUR LOADER MODULE IS NAMED DIFFERENTLY
-from plot_scripts.load_results import load_results_dataframe
+from load_results import load_results_dataframe
+from matplotlib.lines import Line2D
 
 # ----------------------------
 # Config
@@ -26,6 +26,10 @@ SEMI_ANOMALOUS = 10
 
 # Which evaluation columns to plot
 EVALS = ["exp_based", "manual_based"]
+EVALS_LABELS = {
+    "exp_based": "Experiment-Label-Based",
+    "manual_based": "Manually-Labeled",
+}
 
 # Latent dimensions to show as 7 subplots
 LATENT_DIMS = [32, 64, 128, 256, 512, 768, 1024]
@@ -188,7 +192,7 @@ def plot_grid_from_df(
     Create a 2x4 grid of subplots, one per latent dim; 8th panel holds legend.
     kind: 'roc' or 'prc'
     """
-    fig_title = f"{kind.upper()} — {eval_type} (semi = {semi_normals}/{semi_anomalous})"
+    fig_title = f"{kind.upper()} — {EVALS_LABELS[eval_type]} (Semi-Labeling Regime = {semi_normals}/{semi_anomalous})"
     fig, axes = _ensure_dim_axes(fig_title)
 
     # plotting order & colors
@@ -213,7 +217,7 @@ def plot_grid_from_df(
         if i >= 7:
             break  # last slot reserved for legend
         ax = axes[i]
-        ax.set_title(f"latent_dim = {dim}")
+        ax.set_title(f"Latent Dim. = {dim}")
         ax.grid(True, alpha=0.3)
 
         if kind == "roc":