标签:Convolutional set implementation Graph PyTorch Modification Networks
Modification of Graph Convolutional Networks in PyTorch
代码见GitHub:Modification-of-Graph-Convolutional-Networks-in-PyTorch
introduction
Modification of Graph Convolutional Networks in PyTorch, visualization of test set results was added in this version with t-SNE algorithm.
Thanks to the original author Thomas Kipf for the open source code, which provided me with the opportunity to learn.
The visualization results are shown below:
In addition, the division of the training set in this project is not the same as the division method in the paper. In the paper, 4 samples are selected from each type of sample to form the training set, while 100 samples are randomly selected as the training set in the project.
PyTorch implementation of Graph Convolutional Networks (GCNs) for semi-supervised classification [1].
For a high-level introduction to GCNs, see: Thomas Kipf, Graph Convolutional Networks (2016)
Note: There are subtle differences between the TensorFlow implementation in https://github.com/tkipf/gcn and this PyTorch re-implementation. This re-implementation serves as a proof of concept and is not intended for reproduction of the results reported in [1].
This implementation makes use of the Cora dataset from [2].
Installation
python setup.py install
Requirements
- PyTorch 0.4 or 0.5
- Python 2.7 or 3.6
Usage
python train.py
The following optional hyperparameters can be set before running:
- no_cuda:Disables CUDA training.
- fastmode:Validate during training pass.
- seed:Random seed.
- epochs:Number of epochs to train.
- lr:Initial learning rate.
- weight_decay:Weight decay (L2 loss on parameters).
- hidden:Number of hidden units.
- dropout:Dropout rate (1 - keep probability).
References
[1] Kipf & Welling, Semi-Supervised Classification with Graph Convolutional Networks, 2016
[2]Sen et al., Collective Classification in Network Data, AI Magazine 2008
qq_41683065 发布了121 篇原创文章 · 获赞 43 · 访问量 2万+ 私信 关注标签:Convolutional,set,implementation,Graph,PyTorch,Modification,Networks 来源: https://blog.csdn.net/qq_41683065/article/details/104173695
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。