From 09f5ce6600494d99c97c97b9522fee30e31136ae Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 15 Oct 2024 17:13:43 +0200
Subject: [PATCH] datasets
---
dataloader.py | 19 ++++++++++---------
1 file changed, 10 insertions(+), 9 deletions(-)
diff --git a/dataloader.py b/dataloader.py
index 165e89c..9bfeb25 100644
--- a/dataloader.py
+++ b/dataloader.py
@@ -195,12 +195,13 @@ class Intentsity_Dataset(Dataset):
return torch.tensor(self.seq[idx]), torch.tensor([self.energy[idx]]).float(), torch.tensor(
self.precursor_charge[idx]), torch.tensor(self.intensity[idx]).float()
-storage = H5ToStorage('database/traintest_hcd.hdf5')
-storage.make_npy_file('data/intensity/sequence_train.npy','sequence_integer')
-storage.make_npy_file('data/intensity/intensity_train.npy', 'intensities_raw')
-storage.make_npy_file('data/intensity/precursor_charge_train.npy', 'precursor_charge_onehot')
-
-storage = H5ToStorage('database/holdout_hcd.hdf5')
-storage.make_npy_file('data/intensity/sequence_holdout.npy','sequence_integer')
-storage.make_npy_file('data/intensity/intensity_holdout.npy', 'intensities_raw')
-storage.make_npy_file('data/intensity/precursor_charge_holdout.npy', 'precursor_charge_onehot')
+if __name__ =='__main__' :
+ storage = H5ToStorage('database/traintest_hcd.hdf5')
+ storage.make_npy_file('data/intensity/sequence_train.npy','sequence_integer')
+ storage.make_npy_file('data/intensity/intensity_train.npy', 'intensities_raw')
+ storage.make_npy_file('data/intensity/precursor_charge_train.npy', 'precursor_charge_onehot')
+
+ storage = H5ToStorage('database/holdout_hcd.hdf5')
+ storage.make_npy_file('data/intensity/sequence_holdout.npy','sequence_integer')
+ storage.make_npy_file('data/intensity/intensity_holdout.npy', 'intensities_raw')
+ storage.make_npy_file('data/intensity/precursor_charge_holdout.npy', 'precursor_charge_onehot')
--
GitLab