GeneDR

This is a project about drug-disease link prediction for our paper under review.

Environment Requirement

• pytorch == 1.8.1+cu111
• numpy == 1.21.6
• pandas == 1.3.4
• networkx == 2.6.3

Installation

pip install -r requirements.txt

Data Download

• Dataset 1
• Dataset 2 (Datasets will be uploaded soon.)

Scripts Description

• /model/ ** model.py **
• dataset.py
• utils.py
• main.py

Example to Run

python main.py