README.md 1.03 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
Simple Graph Convolution (SGC)
============

- Paper link: [Simplifying Graph Convolutional Networks](https://arxiv.org/abs/1902.07153)
- Author's code repo: [https://github.com/Tiiiger/SGC](https://github.com/Tiiiger/SGC). 

Dependencies
------------
- PyTorch 0.4.1+
- requests

``bash
pip install torch requests
``

Codes
-----
The folder contains an implementation of SGC (`sgc.py`).
Tianyi's avatar
Tianyi committed
19
`sgc_reddit.py` contains an example of training SGC on the reddit dataset.
20
21
22
23
24
25

Results
-------

Run with following (available dataset: "cora", "citeseer", "pubmed")
```bash
Chao Ma's avatar
Chao Ma committed
26
27
28
python3 sgc.py --dataset cora --gpu 0
python3 sgc.py --dataset citeseer --weight-decay 5e-5 --n-epochs 150 --bias --gpu 0
python3 sgc.py --dataset pubmed --weight-decay 5e-5 --bias --gpu 0
29
```
Tianyi's avatar
Tianyi committed
30
31
32
33
Run the following command to train on the reddit dataset.
```bash
python sgc_reddit.py --gpu 0
```
34
35
36
37
38
39

On NVIDIA V100

* cora: 0.819 (paper: 0.810), 0.0008s/epoch
* citeseer: 0.725 (paper: 0.719), 0.0008s/epoch
* pubmed: 0.788 (paper: 0.789), 0.0007s/epoch
Tianyi's avatar
Tianyi committed
40
* reddit: 0.947 (paper: 0.949), 0.6872s in total