From d0065a929dbfbfe9a7aa988fec8e637e7cce4f9d Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Fri, 21 Mar 2025 09:22:46 +0100
Subject: [PATCH] extracting peptide set from txt
---
main.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/main.py b/main.py
index a9aa6e4..d2227c6 100644
--- a/main.py
+++ b/main.py
@@ -215,7 +215,7 @@ def run_duo(args):
def make_prediction_duo(model, data, f_name):
y_pred = []
y_true = []
-
+ print('Building confusion matrix')
# iterate over test data
for imaer,imana, label in data:
label = label.long()
@@ -238,6 +238,8 @@ def make_prediction_duo(model, data, f_name):
cf_matrix = confusion_matrix(y_true, y_pred)
df_cm = pd.DataFrame(cf_matrix[:, None], index=[i for i in classes],
columns=[i for i in classes])
+
+ print('Saving Confusion Matrix')
plt.figure(figsize=(14, 9))
sn.heatmap(df_cm, annot=True)
plt.savefig(f_name)
--
GitLab