Table of Contents

Shortcuts

Retrieval Mean Average Precision (MAP)¶

Module Interface¶

torchmetrics.RetrievalMAP[source]: alias of torchmetrics.retrieval._deprecated._RetrievalMAP

Functional Interface¶

torchmetrics.functional.retrieval_average_precision(preds, target, top_k=None)[source]

Wrapper for deprecated import.

>>> from torch import tensor
>>> preds = tensor([0.2, 0.3, 0.5])
>>> target = tensor([True, False, True])
>>> _retrieval_average_precision(preds, target)
tensor(0.8333)

Return type: Tensor