README.md 849 Bytes
Newer Older
hbsun2113's avatar
hbsun2113 committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
Inductive Representation Learning on Large Graphs (GraphSAGE)
============

- Paper link: [http://papers.nips.cc/paper/6703-inductive-representation-learning-on-large-graphs.pdf](http://papers.nips.cc/paper/6703-inductive-representation-learning-on-large-graphs.pdf)
- Author's code repo: [https://github.com/williamleif/graphsage-simple](https://github.com/williamleif/graphsage-simple). Note that the original code is 
simple reference implementation of GraphSAGE.

Requirements
------------
- requests

``bash
pip install requests
``


Results
-------

Run with following (available dataset: "cora", "citeseer", "pubmed")
```bash
22
python3 train_full.py --dataset cora --gpu 0
hbsun2113's avatar
hbsun2113 committed
23
24
```

25
26
27
* cora: ~0.8330 
* citeseer: ~0.7110
* pubmed: ~0.7830
28
29
30
31
32
33
34
35


Train w/ mini-batch sampling (on the Reddit dataset)
```bash
python3 train_sampling.py
```

Accuracy: 0.9504