wangleiofficial commited on
Commit
7d78b04
·
1 Parent(s): afc08e9

Upload README.md

Browse files
Files changed (1) hide show
  1. README.md +26 -0
README.md ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ## NeuroPred-PLM: an interpretable and robust model for prediction of neuropeptides by protein language model
2
+
3
+ ### Requirements
4
+ To install requirements:
5
+
6
+ ```
7
+ pip install git+https://github.com/ISYSLAB-HUST/NeuroPred-PLM.git
8
+ ```
9
+ ### Usage
10
+
11
+ ```
12
+ import torch
13
+ from NeuroPredPLM.predict import predict
14
+ data = [
15
+ ("peptide_1", "IGLRLPNMLKF"),
16
+ ("peptide_2", "QAAQFKVWSASELVD"),
17
+ ("peptide_3","LRSPKMMHKSGCFGRRLDRIGSLSGLGCNVLRKY")
18
+ ]
19
+
20
+ device = "cuda" if torch.cuda.is_available() else "cpu"
21
+ neuropeptide_pred = predict(data,device)
22
+ # {peptide_id:[Type:int(1->neuropeptide,0->non-neuropeptide), attention score:nd.array]}
23
+ ```
24
+
25
+ ### Contact
26
+ If you have any questions, comments, or would like to report a bug, please file a Github issue or contact me at wanglei94@hust.edu.cn.