utils.py 文件源码

python
阅读 27 收藏 0 点赞 0 评论 0

项目:SeqMatchSeq 作者: pcgreat 项目源码 文件源码
def MRR(ground_label: torch.FloatTensor, predict_label: torch.FloatTensor):
    mrr = 0
    map_idx = 0
    extracted = {}

    for idx_, glab in enumerate(ground_label):
        if ground_label[idx_] != 0:
            extracted[idx_] = 1

    val, key = torch.sort(predict_label, 0, True)
    for i, idx_ in enumerate(key):
        if idx_ in extracted:
            mrr = 1.0 / (i + 1)
            break

    assert (mrr != 0)
    return mrr
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号