f4beaac559e00a3676d942dc7e8fea69efc01cfe,catalyst/metrics/hitrate.py,,hitrate,#Any#Any#Any#,8

Before Change

    _, indices_for_sort = outputs.sort(descending=True, dim=-1)
    true_sorted_by_preds = torch.gather(
        targets, dim=-1, index=indices_for_sort
    )
    true_sorted_by_pred_shrink = true_sorted_by_preds[:, :k]
    hits = torch.sum(true_sorted_by_pred_shrink, dim=1) / k
    return hits


__all__ = ["hitrate"]

After Change

        hitrate_at_k (List[torch.Tensor]):
            the hit rate score
    
    results = []

    targets_sort_by_outputs = process_recsys_components(outputs, targets)
    for k in topk:
        k = min(outputs.size(1), k)
        hits_score = torch.sum(targets_sort_by_outputs[:, :k], dim=1) / k
        results.append(torch.mean(hits_score))

    return results


__all__ = ["hitrate"]

Instances