From 13d0c477a1623dc76cc10cd9fe9ae3f7ed0065ce Mon Sep 17 00:00:00 2001 From: Old-Shatterhand Date: Mon, 11 Nov 2024 09:30:04 +0100 Subject: [PATCH] Fixes in experiment visualization --- experiments/DTI/visualize.py | 8 ++++---- experiments/MPP/visualize.py | 4 ++-- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/experiments/DTI/visualize.py b/experiments/DTI/visualize.py index 3fa7b1a..4d19486 100644 --- a/experiments/DTI/visualize.py +++ b/experiments/DTI/visualize.py @@ -562,10 +562,10 @@ def plot(full_path: Path): print("Plot 3x3") plot_3x3(full_path, data) - print("Plot cold drug") - plot_cold_drug(full_path, data) - print("Plot cold prot") - plot_cold_prot(full_path, data) + #print("Plot cold drug") + #plot_cold_drug(full_path, data) + #print("Plot cold prot") + #plot_cold_prot(full_path, data) if __name__ == '__main__': diff --git a/experiments/MPP/visualize.py b/experiments/MPP/visualize.py index addd48c..527646f 100644 --- a/experiments/MPP/visualize.py +++ b/experiments/MPP/visualize.py @@ -138,9 +138,9 @@ def plot_double(full_path: Path, names: List[str]) -> None: i_tr, i_te, c_tr, c_te = embed(full_path, name.lower()) plot_embeds(ax[i][0], i_tr, i_te, "Random baseline (I1)", legend=True) - set_subplot_label(ax[i][0], fig, "A") + set_subplot_label(ax[i][0], fig, chr(ord("A") + 3 * i)) plot_embeds(ax[i][1], c_tr, c_te, "DataSAIL split (S1)") - set_subplot_label(ax[i][1], fig, "B") + set_subplot_label(ax[i][1], fig, chr(ord("B") + 3 * i)) plt.tight_layout() plt.savefig(full_path / "plots" / f"{names[0]}_{names[1]}.png") plt.show()