Closed XLJZT closed 9 months ago
添加头预测的部分代码,通过直接讲关系数量相加在 id 上 修改 model 中的 fc 层的输入,以及模型计算时取消将卷积特征相加 经过调整,模型测试集结果可对齐论文效果(在误差范围内,FB15K 数据集上,mrr = 0.348,相差 0.002)
添加头预测的部分代码,通过直接讲关系数量相加在 id 上 修改 model 中的 fc 层的输入,以及模型计算时取消将卷积特征相加 经过调整,模型测试集结果可对齐论文效果(在误差范围内,FB15K 数据集上,mrr = 0.348,相差 0.002)