DeepSEQreen_fast_build / requirements.txt
libokj's picture
Update requirements.txt
89ea2d0
raw
history blame
422 Bytes
# deep learning
# Install with conda
# torch>=2.0
# torch_geometric
lightning>=2.0
torchmetrics
transformers
scikit-learn
dgl
dgllife
# configuration and utilities
hydra-core
pandas
tensorboard
# model-specific requirements
subword-nmt # MolTrans
tape_proteins # TransformerCPI2
fair-esm # CoaDTI-pro
# apex # recommended by TAPE-BERT
# feature extraction
rdkit
mordred
networkx>=3.0
mhfp
panel
plotly
seaborn
# tmap