/GCN_PyTorch

pytorch实现GCN代码的中文注释

Primary LanguagePython

GCN_PyTorch

pytorch实现GCN代码的中文注释

参考论文《Semi-Supervised Classification with Graph Convolutional Networks》。

代码注释

最近刚开始学习GCN和PyTorch，所以在学过程中对代码添加了中文注释来加深自己的理解。

源代码来自论文作者Thomas Kipf提供的PyTorch版代码pygcn实现。

源代码工程主要包括五个文件：

__init__.py
layers.py
utils.py
models.py
train.py

参考资料

[1] tkipf/pygcn

[2] Kipf & Welling, Semi-Supervised Classification with Graph Convolutional Networks, 2016