Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/huangwb/AS-GCN

tensorflow-as-gcn
https://github.com/huangwb/AS-GCN

Last synced: 8 days ago
JSON representation

tensorflow-as-gcn

Awesome Lists containing this project

README

        

# AS-GCN in Tensorflow

We provide Tensorflow implementations for the paper "Adaptive Sampling Towards Fast Graph Representation Learning". You need to install Tensorflow and other related python packages, e.g. scypy, networkx, etc. Our code is based on the orginal GCN framework, and takes inspirations from GraphSAGE and FastGCN. The core of this code is that we separate the sampling (i.e. sampler) and propagation (i.e. propagator) processes, both of which are implemented by tensorflow.

Please note that it is possible that the results by this code would be slightly different to those reported in the paper due to the random noise and some post-modifications of the code.

## Data Preparation
We provide the datasets including cora, citeseer, pubmed. For reddit, you need to download it from http://snap.stanford.edu/graphsage/ and then transfer it to the correct format using transfer_Graph.py.

## Run the Code

For cora, citeseer and pubmed, try
```
python run_pubmed.py --dataset dataset_name
```

## Citation
If you find this code useful for your research, please cite the paper:

```
@inproceedings{huang2018adapt,
title={Adaptive Sampling Towards Fast Graph Representation Learning},
author={Huang, Wenbing and Zhang, Tong and Rong, Yu and Huang, Junzhou},
booktitle={Advances in Neural Information Processing Systems (NIPS)},
pages={},
year={2018}
}
```