备注
您正在阅读 MMClassification 0.x 版本的文档。MMClassification 0.x 会在 2022 年末被切换为次要分支。建议您升级到 MMClassification 1.0 版本,体验更多新特性和新功能。请查阅 MMClassification 1.0 的安装教程、迁移教程以及更新日志。
mmcls.core.average_precision¶
- mmcls.core.average_precision(pred, target)[源代码]¶
Calculate the average precision for a single class.
AP summarizes a precision-recall curve as the weighted mean of maximum precisions obtained for any r’>r, where r is the recall:
\[\text{AP} = \sum_n (R_n - R_{n-1}) P_n\]Note that no approximation is involved since the curve is piecewise constant.
- 参数
pred (np.ndarray) – The model prediction with shape (N, ).
target (np.ndarray) – The target of each prediction with shape (N, ).
- 返回
a single float as average precision value.
- 返回类型