Functionhits_at计算Hits@n = #(rank_i <= n) / N 例如,predicts=[[2, 0, 1], [2, 1, 0], [1, 0, 2]], answers=[1, 2, 0], ranks=[3, 1, 2], Hits@2=2/3 :param n: int 要计
gnn/utils/metrics.py:61
Functionmean_reciprocal_rank计算平均倒数排名(MRR) = sum(1 / rank_i) / N,如果答案不在预测结果中,则排名按∞计算 例如,predicts=[[2, 0, 1], [2, 1, 0], [1, 0, 2]], answers=[1, 2, 8], ranks=[3, 2, ∞], MR
gnn/utils/metrics.py:47