Switch to unified view

a/README.md b/README.md
1
# Knowledge Graph Neural Network
1
# Knowledge Graph Neural Network
2
This is our implementation for the paper
2
This is our implementation for the paper
3
> Xuan Lin, Zhe Quan, Zhi-Jie Wang, Tengfei Ma and Xiangxiang Zeng. KGNN: Knowledge Graph Neural Network for Drug-Drug Interaction Prediction. IJCAI' 20 accepted.
3
 Xuan Lin, Zhe Quan, Zhi-Jie Wang, Tengfei Ma and Xiangxiang Zeng. KGNN: Knowledge Graph Neural Network for Drug-Drug Interaction Prediction. IJCAI' 20 accepted.
4
4
5
<img align="center" src="Figure1.png">
5
<img align="center" src="https://github.com/xzenglab/KGNN/blob/master/Figure1.png?raw=true">
6
Figure 1 shows the overview of KGNN. It takes the parsed DDI matrix and knowledge graph obtained from preprocessing of dataset as the input. It outputs the interaction value for the drug-drug pair. 
6
Figure 1 shows the overview of KGNN. It takes the parsed DDI matrix and knowledge graph obtained from preprocessing of dataset as the input. It outputs the interaction value for the drug-drug pair. 
7
7
8
# Requirement
8
# Requirement
9
To run the code, you need the following dependencies:
9
To run the code, you need the following dependencies:
10
* Python == 3.6.6
10
* Python == 3.6.6
11
* Keras == 2.3.0
11
* Keras == 2.3.0
12
* Tensorflow == 1.13.1
12
* Tensorflow == 1.13.1
13
* scikit-learn == 0.22
13
* scikit-learn == 0.22
14
14
15
# Installation
15
# Installation
16
You can create a virtual environment using [conda](https://github.com/conda/conda).
16
You can create a virtual environment using [conda](https://github.com/conda/conda).
17
```bash
17
```bash
18
conda create -n kgnn python=3.6.6  
18
conda create -n kgnn python=3.6.6  
19
source activate kgnn  
19
source activate kgnn  
20
git clone https://github.com/xzenglab/KGNN.git  
20
git clone https://github.com/xzenglab/KGNN.git  
21
cd KGNN  
21
cd KGNN  
22
pip install -r requirement.txt  
22
pip install -r requirement.txt  
23
```
23
```
24
24
25
# Dataset
25
# Dataset
26
We just provide the preprocessed KG from KEGG-drug dataset owing to the size limited. And you can directly download the original DrugBank dataset ([V5.1.4](https://www.drugbank.ca/releases/5-1-4)). Note that the construction of KG please refer to [Bio2RDF](https://github.com/bio2rdf/bio2rdf-scripts/wiki) tool in detail.
26
We just provide the preprocessed KG from KEGG-drug dataset owing to the size limited. And you can directly download the original DrugBank dataset ([V5.1.4](https://www.drugbank.ca/releases/5-1-4)). Note that the construction of KG please refer to [Bio2RDF](https://github.com/bio2rdf/bio2rdf-scripts/wiki) tool in detail.
27
27
28
# Usage
28
# Usage
29
```bash
29
```bash
30
python run.py
30
python run.py
31
```
31
```
32
32
33
# Citation
33
# Citation
34
```bash
34
```bash
35
@inproceedings{ijcai2020-380,  
35
@inproceedings{ijcai2020-380,  
36
  title     = {KGNN: Knowledge Graph Neural Network for Drug-Drug Interaction Prediction},  
36
  title     = {KGNN: Knowledge Graph Neural Network for Drug-Drug Interaction Prediction},  
37
  author    = {Lin, Xuan and Quan, Zhe and Wang, Zhi-Jie and Ma, Tengfei and Zeng, Xiangxiang},  
37
  author    = {Lin, Xuan and Quan, Zhe and Wang, Zhi-Jie and Ma, Tengfei and Zeng, Xiangxiang},  
38
  booktitle = {Proceedings of the Twenty-Ninth International Joint Conference on Artificial Intelligence, {IJCAI-20}},  
38
  booktitle = {Proceedings of the Twenty-Ninth International Joint Conference on Artificial Intelligence, {IJCAI-20}},  
39
  publisher = {International Joint Conferences on Artificial Intelligence Organization},               
39
  publisher = {International Joint Conferences on Artificial Intelligence Organization},               
40
  editor    = {Christian Bessiere},     
40
  editor    = {Christian Bessiere},     
41
  pages     = {2739--2745},    
41
  pages     = {2739--2745},    
42
  year      = {2020},  
42
  year      = {2020},  
43
  month     = {7},  
43
  month     = {7},  
44
  note      = {Main track},  
44
  note      = {Main track},  
45
  doi       = {10.24963/ijcai.2020/380},  
45
  doi       = {10.24963/ijcai.2020/380},  
46
  url       = {https://doi.org/10.24963/ijcai.2020/380},  
46
  url       = {https://doi.org/10.24963/ijcai.2020/380},  
47
}  
47
}  
48
```
48
```
49
49
50
For any clarification, comments, or suggestions please create an issue or contact [Jacklin](Jack_lin@hnu.edu.cn).
50
For any clarification, comments, or suggestions please create an issue or contact [Jacklin](Jack_lin@hnu.edu.cn).