
GCN_链接预测:基于PyTorch的GCN链接预测方法
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目采用PyTorch实现了一种基于图卷积网络(GCN)的链接预测方法。通过分析节点特征与结构信息,有效提升了复杂网络中潜在连接关系的预测精度。
该项目旨在使用PyTorch上的GCN模型进行专利CPC节点的链接预测。为了实现这一目标,采用了Kipf提出的通用GCN架构,并从移动支付行业爬取相关专利数据,在Google专利高级搜索中通过关键词“移动支付”获取专利号。
利用获得的专利号检索所有相关信息后,构建了邻接矩阵和特征矩阵,然后删除不必要的链接并将数据划分为训练集与验证集。接着,通过对GCN图层进行操作来生成新的节点特征,并计算各节点对之间的相似度。通过最小化带有标签信息的损失函数并更新权重的方式完成模型训练。
项目执行时使用以下命令:
- `python crawling.py`
- `python removelinks.py`
- `python features.py`
- `python train.py`
最佳训练轮次为44至46。参考文献包括kenyonke/LinkPredictionGCN和tkipf/pygcn。
全部评论 (0)
还没有任何评论哟~


