libokj commited on
Commit
f9ad405
·
1 Parent(s): eb0f734

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +7 -4
requirements.txt CHANGED
@@ -1,12 +1,13 @@
1
  # deep learning
2
  # Install with conda
3
- # torch>=2.0
4
- # torch_geometric
5
- lightning>=2.0
6
  torchmetrics
7
  transformers
8
  scikit-learn
9
  dgl
 
10
  dgllife
11
 
12
  # configuration and utilities
@@ -14,17 +15,19 @@ hydra-core
14
  pandas
15
  swifter
16
  tensorboard
 
17
 
18
  # model-specific requirements
19
  subword-nmt # MolTrans
20
  tape_proteins # TransformerCPI2
 
21
  fair-esm # CoaDTI-pro
22
  # apex # recommended by TAPE-BERT
23
 
24
  # feature extraction
25
  rdkit
26
  mordred
27
- networkx>=3.0
28
  mhfp
29
  panel
30
  plotly
 
1
  # deep learning
2
  # Install with conda
3
+ torch >= 2.1
4
+ torch_geometric
5
+ lightning >= 2.1
6
  torchmetrics
7
  transformers
8
  scikit-learn
9
  dgl
10
+ dglgo
11
  dgllife
12
 
13
  # configuration and utilities
 
15
  pandas
16
  swifter
17
  tensorboard
18
+ biopython
19
 
20
  # model-specific requirements
21
  subword-nmt # MolTrans
22
  tape_proteins # TransformerCPI2
23
+ gensim
24
  fair-esm # CoaDTI-pro
25
  # apex # recommended by TAPE-BERT
26
 
27
  # feature extraction
28
  rdkit
29
  mordred
30
+ networkx >= 3.2
31
  mhfp
32
  panel
33
  plotly