From 5fc9f56378a377ef7ebf29ed2ca8d47b5c7a022b Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 11 Feb 2025 14:37:30 +0100
Subject: [PATCH] dataset pred for diann
---
diann_lib_processing.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/diann_lib_processing.py b/diann_lib_processing.py
index dbf35f0..51ba849 100644
--- a/diann_lib_processing.py
+++ b/diann_lib_processing.py
@@ -72,7 +72,8 @@ if __name__ =='__main__':
model.load_state_dict(torch.load(args.model_weigh, weights_only=True))
- data_test = load_data(data_source=args.dataset_test, batch_size=args.batch_size, length=30, mode=args.split_test,
+ print(args.dataset_test)
+ data_test = load_data(data_source='data/spectral_lib/data_uniprot_base.csv', batch_size=args.batch_size, length=30, mode=args.split_test,
seq_col=args.seq_test)
predict(data_test, model, args.output)
\ No newline at end of file
--
GitLab