MapleL
/
ctm-dqn
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

This commit is contained in:
Zihan Ye 2026-04-10 01:24:40 +08:00
parent fe60757f0b
commit ac3709a7a8
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
utils/episode_artifacts.py
View File

@ -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): 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_title(f"{title_prefix} {title}")
ax.set_xlabel("Decision Step") ax.set_xlabel("Decision Step")
ax.set_ylabel("Controlled Edge") ax.set_ylabel("Controlled Edge")