避免绘制热力图时自动启用模糊

2026-04-10 01:24:40 +08:00 · 2026-04-10 01:24:40 +08:00 · ac3709a7a8
parent fe60757f0b
commit ac3709a7a8
1 changed files with 8 additions and 1 deletions
--- a/utils/episode_artifacts.py
+++ b/utils/episode_artifacts.py
@ -110,7 +110,14 @@ def _plot_episode_heatmap(path: str, edge_rows: Sequence[Dict], edge_ids: Sequen
    ]

    for ax, (grid, cmap, title) in zip(axes, plots):
-        image = ax.imshow(np.ma.masked_invalid(grid), aspect="auto", origin="lower", cmap=cmap)
+        image = ax.imshow(
+            np.ma.masked_invalid(grid),
+            aspect="auto",
+            origin="lower",
+            cmap=cmap,
+            interpolation="nearest",
+            resample=False,
+        )
        ax.set_title(f"{title_prefix} {title}")
        ax.set_xlabel("Decision Step")
        ax.set_ylabel("Controlled Edge")