liuxiyang641 / RAGAT

RAGAT: Relation Aware Graph Attention Network for Knowledge Graph Completion
56 stars 16 forks source link

学长您好,请问代码中这个gat稀疏矩阵方法SpecialSpmmFinal()执行后具体是什么效果呢 #6

Closed Jim-Hou closed 2 years ago

Jim-Hou commented 2 years ago

我在阅读您的论文时,理解您是在CompGCN的基础上,给关系、尾节点cross过后的嵌入通过软注意力学习了一个权重,再按权重聚合到头节点上。请问这部分思想的代码实现位置是agg_multi_head这里吗,这个函数中的写法我没有太看懂,主要是对SpecialSpmmFinal()不太理解,可否大致解释一下呢? 另外,我看您在FB-237的MRR值达到了0.365,使用的是二头。我没有复现您的模型进行测试,想问您如果去掉多头,MRR值大约是多少呢? 感谢!