From bea45844ee7da4f2d6bc4e322d3cf0de08722b5c Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Fri, 18 Oct 2024 15:49:34 +0200
Subject: [PATCH] datasets
---
common_dataset.py | 13 +++++++------
1 file changed, 7 insertions(+), 6 deletions(-)
diff --git a/common_dataset.py b/common_dataset.py
index 462db86..17dc21a 100644
--- a/common_dataset.py
+++ b/common_dataset.py
@@ -165,17 +165,18 @@ def load_data(path_train, path_val, path_test, batch_size, length, pad=False, co
return train_loader, val_loader, test_loader
if __name__ =='__main__' :
- irt_train = np.load('data/intensity/irt_train.npy')
- 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')
+ pass
+ # irt_train = np.load('data/intensity/irt_train.npy')
+ # 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/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_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_test.to_pickle('database/data_prosit_merged_holdout.pkl')
\ No newline at end of file
--
GitLab