Update run_support.py

run_support.py

@@ -3,6 +3,25 @@ import matplotlib.pyplot as plt
 
 from metrics import compare_all_fast_with_curves
 
+COLORS = {
+    "ALSA": "C0",   # blue
+    "LISA": "C3",   # red
+}
+
+METRICS = {
+    "MSE":               ["mse", "MSE", "mse_mean"],
+    "Spectral (JS/KL)":  ["spectral_kl", "spectral_js", "spectral_div", "spec_kl", "spectral"],
+    "ACF-MSE":           ["acf_mse", "acf_l2", "acf_err", "acf"],
+    "MMD$^2$ (RFF)":     ["mmd2_rff", "mmd_rff", "mmd2", "mmd"],
+}
+
+styles = {
+    "Truth": dict(color="k",  lw=2.3, ls="-"),
+    "NLSA":  dict(color="0.2", lw=2.0, ls="--"),
+    "ALSA":  dict(color=COLORS["ALSA"], lw=2.0, ls="-.", alpha=0.95),
+    "LISA":  dict(color=COLORS["LISA"], lw=2.1, ls=":", alpha=0.95),
+}
+
 def standardize_global(F_tX: np.ndarray, eps: float = 1e-12):
     mu = F_tX.mean(axis=0, keepdims=True)
     sd = F_tX.std(axis=0, keepdims=True) + eps