1 topk_accuracy_from_probs
rri_metrics.topk_accuracy_from_probs(probs, labels, *, top_k)Compute top-k accuracy from class probabilities.
1.1 Parameters
| Name | Type | Description | Default |
|---|---|---|---|
| probs | Tensor | Tensor["N K"] class probabilities. |
required |
| labels | Tensor | Tensor["N"] integer class labels. |
required |
| top_k | int | Number of highest-probability classes to consider. | required |
1.2 Returns
| Name | Type | Description |
|---|---|---|
| Tensor | Tensor[""] scalar accuracy in [0, 1]. |