From 538eee31132911db4a290a5d3b41b303bfe0fb79 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 15 Oct 2024 13:12:19 +0200
Subject: [PATCH] data merge intensity

---
 common_dataset.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/common_dataset.py b/common_dataset.py
index bab5944..42c4c19 100644
--- a/common_dataset.py
+++ b/common_dataset.py
@@ -170,12 +170,12 @@ seq_train = np.load('data/intensity/sequence_train.npy')
 charge_train = np.load('data/intensity/precursor_charge_train.npy')
 spectra_train = np.load('data/intensity/intensity_train.npy')
 
-# irt_holdout = np.load('data/intensity/collision_irt_holdout.npy')
-# seq_holdout = np.load('data/intensity/sequence_holdout.npy')
-# charge_holdout = np.load('data/intensity/precursor_charge_holdout.npy')
-# spectra_holdout = np.load('data/intensity/intensity_holdout.npy')
+irt_holdout = np.load('data/intensity/collision_irt_holdout.npy')
+seq_holdout = np.load('data/intensity/sequence_holdout.npy')
+charge_holdout = np.load('data/intensity/precursor_charge_holdout.npy')
+spectra_holdout = np.load('data/intensity/intensity_holdout.npy')
 
 dataset_train = pd.DataFrame({'Sequence':list(seq_train), 'Retention time':list(irt_train), 'Charge':list(charge_train), 'Spectra' : list(spectra_train)},index=list(range(6787933)))
 dataset_train.to_pickle('database/data_prosit_merged_train.pkl')
-# dataset_test = pd.DataFrame({'Sequence':list(seq_holdout), 'Retention time':list(irt_holdout), 'Charge':list(charge_holdout), 'Spectra' : list(spectra_holdout)},index=list(range(754215)))
-# dataset_train.to_pickle('database/data_prosit_merged_holdout.pkl')
\ No newline at end of file
+dataset_test = pd.DataFrame({'Sequence':list(seq_holdout), 'Retention time':list(irt_holdout), 'Charge':list(charge_holdout), 'Spectra' : list(spectra_holdout)},index=list(range(754215)))
+dataset_train.to_pickle('database/data_prosit_merged_holdout.pkl')
\ No newline at end of file
-- 
GitLab