shuibinlong / ConvR

The pytorch implementation of ConvR model from "Adaptive Convolution for Multi-Relational Learning"
25 stars 6 forks source link

add head predict and modify model #5

Closed XLJZT closed 9 months ago

XLJZT commented 10 months ago

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