/PyTorch-GNN

The implement of GNN based on Pytorch

Primary LanguagePython

基于PyTorch框架实现图卷积神经网络

项目源代码参考本人Github.

依赖库

  • DGL 0.1.3
  • PyTorch 0.4.1
  • networkX 2.2

过程

1. 利用DGL构建图

2. 构建图卷积神经网络

3. 训练过程

4. 训练过程可视化

result

项目参考:教程