From 0ec4aed4e01c553d3b6387d3aa8e1af0a3d6ace6 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Fri, 20 Sep 2024 11:47:03 +0200
Subject: [PATCH] dataloader fix
---
.idea/LC-MS-RT-prediction.iml | 2 +-
main_custom.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/.idea/LC-MS-RT-prediction.iml b/.idea/LC-MS-RT-prediction.iml
index 82f955b..81c1f05 100644
--- a/.idea/LC-MS-RT-prediction.iml
+++ b/.idea/LC-MS-RT-prediction.iml
@@ -9,7 +9,7 @@
<excludeFolder url="file://$MODULE_DIR$/.venv" />
<excludeFolder url="file://$MODULE_DIR$/.venv2" />
</content>
- <orderEntry type="inheritedJdk" />
+ <orderEntry type="jdk" jdkName="Python 3.11 (LC-MS-RT-prediction)" jdkType="Python SDK" />
<orderEntry type="sourceFolder" forTests="false" />
</component>
</module>
\ No newline at end of file
diff --git a/main_custom.py b/main_custom.py
index ea21899..1118c9a 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -202,7 +202,7 @@ def main(args):
path_test=args.dataset_test,
batch_size=args.batch_size, length=25, pad = False, convert=True, vocab='iapuc')
elif args.forward == 'rt':
- data_train, data_val, data_test = dataloader.load_data(data_sources=args.dataset_train,
+ data_train, data_val, data_test = dataloader.load_data(data_sources=[args.dataset_train,args.dataset_val,args.dataset_test],
batch_size=args.batch_size, length=25)
print('\nData loaded')
--
GitLab