北京其它兼职程序员外包

需求描述

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