how to calculate Mean Rank in metric.py

Question

Opened this issue 3 years ago · 0 comments

Hello,

I wanna know how to calculate Mean Rank in metric.py like MRR

is this right?
((indices == ground_truth_idx.cuda()).nonzero()[:, 1].float().add(1.0)).sum().item()