A robust drug-target interaction prediction framework with capsule network and transfer learning
This is the code for CapBM-DTI.
- Python 3.x
- numpy
- scikit-learn
- RDKit
- Tensorflow
- keras
To run this code:
python CapBM_DTI.py --dti data/benchmark.txt --protein-descripter bert --drug-descripter MPNN --model-name bert_MPNN_capsule --batch-size 64 -e 1000 -dp data