kiholee/edma-semantic-mac
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fig 5 legend lower right, genie in caption; shorten hybrid label

Browse Source
This commit is contained in:
KiHoLee
2026-08-17 15:42:11 +09:00
parent 2aa8ee16ea
commit 916693e5eb
8 changed files with 9 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
code/replot_all.py
+9 -11
View File
@@ -36,7 +36,7 @@ LBL = {
"todma": "ToDMA-adapted",
"mac": "MAC sum capacity",
"coop": "Full-cooperation bound",
"hybrid": "EDMA + refinement stage",
"hybrid": "EDMA + refinement",
}
@@ -190,20 +190,18 @@ def fig_retrieval():
ax.plot(snr, col(rows, "todma"), "d-.", color="C4",
label=LBL["todma"])
ax.plot(snr, col(rows, "oma"), "v:", color="C1", label=LBL["oma"])
# genie curve overlaps EDMA here; named in the caption, not the legend
ax.plot(snr, col(rows, "genie"), "-", color="gray", lw=1.0,
label=LBL["genie"])
ax.axhline(1.0 / 16, color="gray", ls=":", lw=0.8)
ax.annotate("chance $1/16$", xy=(10.5, 1.0 / 16 + 0.015), fontsize=7,
label="_nolegend_")
ax.axhline(1.0 / 16, xmax=0.48, color="gray", ls=":", lw=0.8)
ax.annotate("chance $1/16$", xy=(1.2, 1.0 / 16 + 0.015), fontsize=7,
color="gray")
ax.set_xlabel("SNR $\\rho$ [dB]")
ax.set_ylabel("Top-1 retrieval accuracy")
ax.set_xlim(snr[0], snr[-1]); ax.set_ylim(0, 1.42)
ax.set_yticks([0.0, 0.2, 0.4, 0.6, 0.8, 1.0])
hs, ls = ax.get_legend_handles_labels()
order = [1, 4, 0, 2, 3] # long labels share column one
ax.legend([hs[i] for i in order], [ls[i] for i in order],
loc="upper center", ncol=2, columnspacing=0.7,
handlelength=1.3, handletextpad=0.5)
ax.set_xlim(snr[0], snr[-1]); ax.set_ylim(0, 1.0)
ax.legend(loc="lower right", borderaxespad=0.3,
handlelength=1.4, handletextpad=0.5,
labelspacing=0.35)
save(fig, "fig_retrieval")