快捷导航

使用图神经网络计算图图神经网络算法

m6145***  √手机认证
使用图神经网络计算图中节点重要性。首先用graphsage计算节点初始嵌入,然后用基于中心度的注意力机制进行改进,就得到了节点的最终嵌入。然后通过一个全连接层计算出节点的重要性分数,使用均方误差进行训练。数据集为常用的引文网络Cora、Citeseer、Pubmed,标签为文章被引用的次数。评估指标为NDCG@K,baseline为pagerank算法,消融实验为不使用注意力机制。
回复

使用道具 举报

发新帖
TA的信息
  • 会员所属: 注册会员
  • 认证信息: 邮箱认证手机认证
  • 微信访问
  • 手机APP