README.md 1.83 KB
Newer Older
Ziyue Huang's avatar
Ziyue Huang committed
1
2
# Stochastic Training for Graph Convolutional Networks

3
4
DEPRECATED!!

Ziyue Huang's avatar
Ziyue Huang committed
5
6
7
8
9
10
11
12
13
14
15
16
17
18
* Paper: [Control Variate](https://arxiv.org/abs/1710.10568)
* Paper: [Skip Connection](https://arxiv.org/abs/1809.05343)
* Author's code: [https://github.com/thu-ml/stochastic_gcn](https://github.com/thu-ml/stochastic_gcn)

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

``bash
pip install torch requests
``

### Neighbor Sampling & Skip Connection
19
20
21
22

#### cora 

Test accuracy ~83% with --num-neighbors 2, ~84% by training on the full graph
Ziyue Huang's avatar
Ziyue Huang committed
23
```
24
DGLBACKEND=pytorch python3 gcn_ns_sc.py --dataset cora --self-loop --num-neighbors 2 --batch-size 1000000 --test-batch-size 1000000
Ziyue Huang's avatar
Ziyue Huang committed
25
26
```

27
28
29
#### citeseer 

Test accuracy ~69% with --num-neighbors 2, ~70% by training on the full graph
Ziyue Huang's avatar
Ziyue Huang committed
30
```
31
DGLBACKEND=pytorch python3 gcn_ns_sc.py --dataset citeseer --self-loop --num-neighbors 2 --batch-size 1000000 --test-batch-size 1000000
Ziyue Huang's avatar
Ziyue Huang committed
32
33
```

34
35
36
#### pubmed 

Test accuracy ~76% with --num-neighbors 3, ~77% by training on the full graph
Ziyue Huang's avatar
Ziyue Huang committed
37
```
38
DGLBACKEND=pytorch python3 gcn_ns_sc.py --dataset pubmed --self-loop --num-neighbors 3 --batch-size 1000000 --test-batch-size 1000000
Ziyue Huang's avatar
Ziyue Huang committed
39
40
41
```

### Control Variate & Skip Connection
42
43
44
45

#### cora 

Test accuracy ~84% with --num-neighbors 1, ~84% by training on the full graph
Ziyue Huang's avatar
Ziyue Huang committed
46
```
47
DGLBACKEND=pytorch python3 gcn_cv_sc.py --dataset cora --self-loop --num-neighbors 1 --batch-size 1000000 --test-batch-size 1000000
Ziyue Huang's avatar
Ziyue Huang committed
48
49
```

50
51
52
#### citeseer 

Test accuracy ~69% with --num-neighbors 1, ~70% by training on the full graph
Ziyue Huang's avatar
Ziyue Huang committed
53
```
54
DGLBACKEND=pytorch python3 gcn_cv_sc.py --dataset citeseer --self-loop --num-neighbors 1 --batch-size 1000000 --test-batch-size 1000000
Ziyue Huang's avatar
Ziyue Huang committed
55
56
```

57
58
59
#### pubmed 

Test accuracy ~77% with --num-neighbors 1, ~77% by training on the full graph
Ziyue Huang's avatar
Ziyue Huang committed
60
```
61
DGLBACKEND=pytorch python3 gcn_cv_sc.py --dataset pubmed --self-loop --num-neighbors 1 --batch-size 1000000 --test-batch-size 1000000
Ziyue Huang's avatar
Ziyue Huang committed
62
63
```