README.md 833 Bytes
Newer Older
1
2
3
Graph Convolutional Networks (GCN)
============

Mufei Li's avatar
Mufei Li committed
4
5
6
- Paper link: [https://arxiv.org/abs/1609.02907](https://arxiv.org/abs/1609.02907)
- Author's code repo: [https://github.com/tkipf/gcn](https://github.com/tkipf/gcn). Note that the original code is 
implemented with Tensorflow for the paper. 
7

8
Dependencies
9
------------
10
- PyTorch 0.4.1+
11
12
13
- requests

``bash
14
pip install torch requests
15
16
``

17
18
Codes
-----
19
The folder contains two implementations of GCN. `gcn.py` uses user-defined
20
21
message and reduce functions. `gcn_spmv.py` uses DGL's builtin functions so
SPMV optimization could be applied.
22

23
24
Results
-------
25

26
27
28
Run with following (available dataset: "cora", "citeseer", "pubmed")
```bash
python gcn_spmv.py --dataset cora --gpu 0
29
30
```

31
32
33
* cora: ~0.810 (0.79-0.83) (paper: 0.815)
* citeseer: 0.707 (paper: 0.703)
* pubmed: 0.792 (paper: 0.790)