Skip to content

Commit 38a6a41

Browse files
authored
Update README.md
1 parent c811c77 commit 38a6a41

File tree

1 file changed

+22
-14
lines changed

1 file changed

+22
-14
lines changed

README.md

+22-14
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,8 @@
11
# H-GCN
2-
## Description
3-
This is the repository for the IJCAI-19 paper [Hierarchical Graph Convolutional Networks for Semi-supervised Node Classification](https://arxiv.org/pdf/1902.06667.pdf).
42

5-
## Requirements
3+
<img src="h_gcn.png" alt="model" style="zoom: 50%;" />
64

7-
- Tensorflow (1.9.0)
8-
- networkx
5+
This is the code for the IJCAI 19 Paper: [Hierarchical Graph Convolutional Networks for Semi-supervised Node Classification](https://dl.acm.org/doi/10.5555/3367471.3367673).
96

107
## Usage
118

@@ -23,15 +20,26 @@ python train.py --dataset citeseer --epochs 200 --early_stopping 60 --coarsen_le
2320
python train.py --dataset pubmed --epochs 250 --early_stopping 1000 --coarsen_level 4 --dropout 0.85 --weight_decay 7e-4 --hidden 30 --node_wgt_embed_dim 8 --seed1 156 --seed2 136
2421
```
2522

26-
## Cite
27-
Please cite our paper if you use this code in your own work:
23+
## Requirements
2824

25+
- Tensorflow (1.9.0)
26+
- networkx
27+
28+
## Citation
29+
30+
Please cite our paper if you use the code:
2931
```
30-
@inproceedings{hgcn_ijcai19,
31-
title = {Hierarchical Graph Convolutional Networks for Semi-supervised Node Classification},
32-
author = {Fenyu Hu and Yanqiao Zhu and Shu Wu and Liang Wang and Tieniu Tan},
33-
booktitle = {Proceedings of the Twenty-Eighth International Joint Conference on Artificial Intelligence, (IJCAI)},
34-
year = {2019},
35-
url = {https://arxiv.org/abs/1902.06667}
32+
@inproceedings{10.5555/3367471.3367673,
33+
author = {Hu, Fenyu and Zhu, Yanqiao and Wu, Shu and Wang, Liang and Tan, Tieniu},
34+
title = {Hierarchical graph convolutional networks for semi-supervised node classification},
35+
year = {2019},
36+
isbn = {9780999241141},
37+
publisher = {AAAI Press},
38+
abstract = {Graph convolutional networks (GCNs) have been successfully applied in node classification tasks of network mining. However, most of these models based on neighborhood aggregation are usually shallow and lack the "graph pooling" mechanism, which prevents the model from obtaining adequate global information. In order to increase the receptive field, we propose a novel deep Hierarchical Graph Convolutional Network (H-GCN) for semi-supervised node classification. H-GCN first repeatedly aggregates structurally similar nodes to hyper-nodes and then refines the coarsened graph to the original to restore the representation for each node. Instead of merely aggregating one- or two-hop neighborhood information, the proposed coarsening procedure enlarges the receptive field for each node, hence more global information can be captured. The proposed H-GCN model shows strong empirical performance on various public benchmark graph datasets, outperforming state-of-the-art methods and acquiring up to 5.9\% performance improvement in terms of accuracy. In addition, when only a few labeled samples are provided, our model gains substantial improvements.},
39+
booktitle = {Proceedings of the 28th International Joint Conference on Artificial Intelligence},
40+
pages = {4532–4539},
41+
numpages = {8},
42+
location = {Macao, China},
43+
series = {IJCAI'19}
3644
}
37-
```
45+
```

0 commit comments

Comments
 (0)