gigl.src.common.utils.hit_rate_at_k

gigl.src.common.utils.eval_metrics.hit_rate_at_k(pos_scores: FloatTensor, neg_scores: FloatTensor, ks: LongTensor) → FloatTensor

Computes Hit Rate @ K metrics for various Ks, evaluating 1+ positives against 1+ negatives.

Args:

pos_scores (torch.FloatTensor): Contains 1 or more positive sample scores. neg_scores (torch.FloatTensor): Contains 1 or more negative sample scores. ks (torch.LongTensor): k-values for which to compute hits.

Returns:

torch.FloatTensor: Hit rates corresponding to the requested ks.