libokj commited on
Commit
c072dc8
·
1 Parent(s): b6be705

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +17 -24
requirements.txt CHANGED
@@ -1,41 +1,34 @@
1
  # deep learning
2
  # Install with conda
3
- torch>=2.1.0
4
- torch_geometric
5
- lightning>=2.1.0
6
  torchmetrics
7
  transformers
8
  scikit-learn
9
  dgl
10
- dglgo
11
  dgllife
12
 
13
  # configuration and utilities
14
  hydra-core
15
- # hydra-colorlog
16
- # hydra-optuna-sweeper
17
- # hydra-submitit-launcher
18
  pandas
19
- swifter
20
- # pre-commit
21
- # rich
22
- # pytest
23
- # tensorboard
24
-
25
- # feature extraction
26
- biopython
27
- rdkit
28
- networkx>=3.2.0
29
- mhfp
30
- # tmap
31
 
32
  # model-specific requirements
33
- # subword-nmt # MONN
34
  tape_proteins # TransformerCPI2
35
- gensim # TransformerCPI
36
  fair-esm # CoaDTI-pro
37
  # apex # recommended by TAPE-BERT
38
 
39
- # data visualization utilities
40
- seaborn
41
- plotly
 
 
 
 
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
13
  hydra-core
14
+ hydra-colorlog
15
+ hydra-optuna-sweeper
16
+ hydra-submitit-launcher
17
  pandas
18
+ pre-commit
19
+ rich
20
+ pytest
21
+ tensorboard
 
 
 
 
 
 
 
 
22
 
23
  # model-specific requirements
24
+ subword-nmt # MolTrans
25
  tape_proteins # TransformerCPI2
 
26
  fair-esm # CoaDTI-pro
27
  # apex # recommended by TAPE-BERT
28
 
29
+ # feature extraction
30
+ rdkit
31
+ mordred
32
+ networkx>=3.0
33
+ mhfp
34
+ # tmap