From 163eb002cd78149329b3a40b7bbb5a429e304fb4 Mon Sep 17 00:00:00 2001
From: lschneider <leo.schneider@univ-lyon1.fr>
Date: Wed, 18 Sep 2024 10:34:25 +0200
Subject: [PATCH] fix save pred
---
main_custom.py | 26 +++++++++++++-------------
1 file changed, 13 insertions(+), 13 deletions(-)
diff --git a/main_custom.py b/main_custom.py
index 340bd98..1f1d51e 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -266,12 +266,12 @@ def save_pred(model, data_val, forward, output_path):
charges.extend(charge.data.cpu().tolist())
true_rt.extend(rt.data.cpu().tolist())
true_int.extend(intensity.data.cpu().tolist())
- data_frame['rt pred'] = pred_rt
- data_frame['seq'] = seqs
- data_frame['pred int'] = pred_int
- data_frame['true rt'] = true_rt
- data_frame['true int'] = true_int
- data_frame['charge'] = charges
+ data_frame['rt pred'] = pred_rt
+ data_frame['seq'] = seqs
+ data_frame['pred int'] = pred_int
+ data_frame['true rt'] = true_rt
+ data_frame['true int'] = true_int
+ data_frame['charge'] = charges
@@ -285,9 +285,9 @@ def save_pred(model, data_val, forward, output_path):
pred_rt.extend(pr_rt.data.cpu().tolist())
seqs.extend(seq.data.cpu().tolist())
true_rt.extend(rt.data.cpu().tolist())
- data_frame['rt pred'] = pred_rt
- data_frame['seq'] = seqs
- data_frame['true rt'] = true_rt
+ data_frame['rt pred'] = pred_rt
+ data_frame['seq'] = seqs
+ data_frame['true rt'] = true_rt
if forward == 'int': #adapted to prosit dataset format
@@ -300,10 +300,10 @@ def save_pred(model, data_val, forward, output_path):
seqs.extend(seq.data.cpu().tolist())
charges.extend(charge.data.cpu().tolist())
true_int.extend(intensity.data.cpu().tolist())
- data_frame['seq'] = seqs
- data_frame['pred int'] = pred_int
- data_frame['true int'] = true_int
- data_frame['charge'] = charges
+ data_frame['seq'] = seqs
+ data_frame['pred int'] = pred_int
+ data_frame['true int'] = true_int
+ data_frame['charge'] = charges
data_frame.to_csv(output_path)
--
GitLab