dragen1860 / Gcn Pytorch
Graph Convolution Network for PyTorch
Stars: ✭ 197
Programming Languages
python
139335 projects - #7 most used programming language
Graph Convolution Network for PyTorch
Pytorch 1.0 with python 3.7.
GCN implementation for paper: Semi-Supervised Classification with Graph Convolutional Networks
Benchmark
| dataset | Citeseea | Cora | Pubmed | NELL |
|---|---|---|---|---|
| GCN(official) | 70.3 | 81.5 | 79.0 | 66.0 |
| This repo. | 81.7 | 78.7 |
Hints:
To achieve 81.7% on Cora dataset, pytorch need 400 epochs.
and it train relatively slower on Pubmed dataset.
HOWTO
python train.py
Screenshot
Note that the project description data, including the texts, logos, images, and/or trademarks,
for each open source project belongs to its rightful owner.
If you wish to add or remove any projects, please contact us at [email protected].