The codebase is built on OpenBioMed. We are endavouring for a future release that incorporates the revised KEDD model.
To reproduce KEDD, install BMKG and put it under assets/kg/
.
Yamanishi08 and BMKG_DTI can download from here. The 2 datasets should put under datasets/dti/
.
To run KEDD for drug-target interaction, use the following command:
bash scripts/dti/train_kedd.sh
Download MoleculeNet datasets here, unzip the file, and put the dataset fold under datasets/dp/
. You can use the following commands from within OpenBioMed/
:
wget http://snap.stanford.edu/gnn-pretrain/data/chem_dataset.zip
unzip chem_dataset.zip
mkdir -p datasets/dp
mv dataset datasets/dp/moleculenet
rm chem_dataset.zip
After downloading and unzipping, you should remove all the processed/
directories of 8 datasets in the dataset/
folder.
To run KEDD for drug property prediction, use the following command:
bash scripts/dp/train_kedd.sh
Luo's dataset is available here. Put them under the datasets/ddi/
folder.
To run KEDD for drug-drug interaction prediction, use the following command:
bash scripts/ddi/train_kedd.sh
The SHS27k and SHS148k datasets are available here. Put them under datasets/ppi/
To run KEDD for protein-protein interaction prediction, use the following command:
bash scripts/ppi/run.sh