/SubKNet

Primary LanguagePython

SubKNet

Code and datasets for "Subgraph-aware graph kernel neural network for link prediction in biological networks"

Developers

Menglu Li (mengluli@foxmail.com) , Zhiwei Wang, Luotao Liu, Xuan Liu, and Wen Zhang (zhangwen@mail.hzau.edu.cn) from College of Informatics, Huazhong Agricultural University.

Tuturial

To get the experiment results by running the script main.py:

python main.py --dataset MDA --types hete --exp_name cv   ###5-fold cross-validation
python main.py --dataset MDA --types hete --exp_name indep   ###independent test

Requirements

  • numpy 1.19.4
  • pandas 0.23.4
  • scipy 1.4.1
  • scikit-learn 0.19.2
  • pytorch 1.4.0
  • python 3.7.2

Contact

Please feel free to contact us if you need any help.