
GCN:使用TensorFlow实现图卷积网络
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目基于TensorFlow框架实现了图卷积神经网络(GCN),适用于节点分类、链接预测等任务。代码开源,易于扩展和应用。
图卷积网络是使用TensorFlow实现的,用于处理图结构数据中的节点分类问题(包括半监督学习任务)。这个项目基于Thomas N.Kipf与Max Welling在ICLR 2017上发表的文章,并且可以在我们的博客文章中找到更详细的解释。要安装,请运行`python setup.py install`命令。
**要求:**
- 张量流版本需大于0.12
该网络的使用演示可以通过以下步骤执行:
```bash
cd gcn
python train.py
```
对于自定义数据集,你需要提供三个矩阵来表示图结构和节点信息:
- N×N大小的邻接矩阵(其中N代表节点的数量)
- N×D大小的特征矩阵(这里D是每个节点所拥有的特征数量)
- 一个N by E大小的二进制标签矩阵(E为类别数)
在`utils.py`中的load_data()函数提供了如何使用这些数据的具体示例。在此演示中,我们加载了引文网络的数据集(包括Cora、Citeseer或Pubmed)。原始数据可以从相关的文献和存储库获取。
请参考相关文档以了解更多信息并开始您的实验。
全部评论 (0)
还没有任何评论哟~


