图神经网络——制作数据集
目前看来，很多图神经网络（GAT/GCN）都有使用cora数据集，但是怎么更换为自己的数据集？——————>难道是换成类似的形式？.x.allx这些，emmmm，曾经我有产生这个疯狂的想法。GAT/GCN的作者在github上已经说明了滴，主要是构建邻接矩阵、特征矩阵以及one-hot形式的标签矩阵。例如，数据集有10242(N)个节点，每个节点拥有6(F)个特征，32©个类别那么你的邻接矩
·
目前看来，很多图神经网络（GAT/GCN）都有使用cora数据集，但是怎么更换为自己的数据集？
——————>难道是换成类似的形式？.x .allx这些，emmmm，曾经我有产生这个疯狂的想法。
GAT/GCN的作者在github上已经说明了滴，主要是构建邻接矩阵、特征矩阵以及one-hot形式的标签矩阵。
例如，数据集有10242(N)个节点，每个节点拥有6(F)个特征，32©个类别
那么你的邻接矩阵维度就是(10242,10242), 特征矩阵就是(10242,6),标签矩阵就是(10242,32).
温馨提示：Pytorch的GAT稀疏版本有问题，因此建议大家使用Tf的。
如果大哥们关于制作数据集还有问题，留言即可。另外我是做图像分割的，小菜鸡欢迎各位大佬交流讨论呀！
魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。
更多推荐
11
0- 0
已为社区贡献1条内容
所有评论(0)