This is a project about drug-disease link prediction for our paper under review.
- pytorch == 1.8.1+cu111
- numpy == 1.21.6
- pandas == 1.3.4
- networkx == 2.6.3
pip install -r requirements.txt
- Dataset 1
- Dataset 2 (Datasets will be uploaded soon.)
- /model/ ** model.py **
- dataset.py
- utils.py
- main.py
python main.py