From f7e65615fafffad7bbec14e3f14d3559b7c9b10e Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 11 Feb 2025 10:04:33 +0100
Subject: [PATCH] dataset pred for diann

---
 diann_lib_processing.py | 34 +++++++++++++++++-----------------
 1 file changed, 17 insertions(+), 17 deletions(-)

diff --git a/diann_lib_processing.py b/diann_lib_processing.py
index 419754d..5228eea 100644
--- a/diann_lib_processing.py
+++ b/diann_lib_processing.py
@@ -54,20 +54,20 @@ if __name__ =='__main__':
     # df_2 = extract_sequence(df).reset_index(drop=True)
     # df_2.to_csv('./data/spectral_lib/data_uniprot_base.csv', index=False)
 
-    df = pd.read_csv('spectral_lib/data_uniprot_base.csv')
-    df['sequence']=df['sequence'].map(lambda x:x.replace('M(UniMod:35)','-OxM-'))
-    df.to_csv('spectral_lib/data_uniprot_base.csv')
-
-    # # args = load_args()
-    #
-    # model = ModelTransformer(encoder_ff=args.encoder_ff, decoder_rt_ff=args.decoder_rt_ff,
-    #                          n_head=args.n_head, encoder_num_layer=args.encoder_num_layer,
-    #                          decoder_rt_num_layer=args.decoder_rt_num_layer, drop_rate=args.drop_rate,
-    #                          embedding_dim=args.embedding_dim, acti=args.activation, norm=args.norm_first)
-    #
-    # 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,
-    #                       seq_col=args.seq_test)
-    #
-    # predict(data_test, model, args.output)
\ No newline at end of file
+    # df = pd.read_csv('spectral_lib/data_uniprot_base.csv')
+    # df['sequence']=df['sequence'].map(lambda x:x.replace('M(UniMod:35)','-OxM-'))
+    # df.to_csv('spectral_lib/data_uniprot_base.csv')
+
+    args = load_args()
+
+    model = ModelTransformer(encoder_ff=args.encoder_ff, decoder_rt_ff=args.decoder_rt_ff,
+                             n_head=args.n_head, encoder_num_layer=args.encoder_num_layer,
+                             decoder_rt_num_layer=args.decoder_rt_num_layer, drop_rate=args.drop_rate,
+                             embedding_dim=args.embedding_dim, acti=args.activation, norm=args.norm_first)
+
+    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,
+                          seq_col=args.seq_test)
+
+    predict(data_test, model, args.output)
\ No newline at end of file
-- 
GitLab