KGE性能指标：MRR，MR，HITS@1，HITS@3，HITS@10

飞机火车巴雷特

已于 2023-07-24 12:02:42 修改

阅读量4.4w

点赞数 33

分类专栏：机器学习文章标签：知识图谱

于 2021-09-12 20:10:51 首次发布

本文链接：https://blog.csdn.net/qq_36158230/article/details/120254381

版权

机器学习专栏收录该内容

33 篇文章 1 订阅

订阅专栏

本文将介绍用于衡量知识图谱嵌入（Knowledge Graph Embedding，KGE）模型性能中最常用的几个指标：MRR，MR，HITS@1，HITS@3，HITS@10。

一、MRR

MRR的全称是Mean Reciprocal Ranking，其中Reciprocal是指“倒数的”的意思。具体的计算方法如下：

$\mathbf{MRR}=\frac{1}{|S|}\sum_{i=1}^{|S|}\frac{1}{rank_i}=\frac{1}{|S|}(\frac{1}{rank_1}+\frac{1}{rank_2}+...+\frac{1}{rank_{|S|}})$

其中 $S$ 是三元组集合， $|S|$ 是三元组个数， $rank_i$ 是指第 $i$ 个三元组的链接预测排名。该指标越大越好。例如，对于三元组(Jack，born_in，Italy)，链接预测的结果可能是

s        p         o            score   rank
Jack   born_in   Ireland        0.789      1
Jack   born_in   Italy          0.753      2  *
Jack   born_in   Germany        0.695      3
Jack   born_in   China          0.456      4
Jack   born_in   Thomas         0.234      5

那么，三元组(Jack，born_in，Italy)的链接预测排名则是2.