备注

您正在阅读 MMClassification 0.x 版本的文档。MMClassification 0.x 会在 2022 年末被切换为次要分支。建议您升级到 MMClassification 1.0 版本，体验更多新特性和新功能。请查阅 MMClassification 1.0 的安装教程、迁移教程以及更新日志。

mmcls.core.precision¶

mmcls.core.precision(pred, target, average_mode='macro', thrs=0.0)[源代码]¶

Calculate precision according to the prediction and target.

参数

pred (torch.Tensor | np.array) – The model prediction with shape (N, C).
target (torch.Tensor | np.array) – The target of each prediction with shape (N, 1) or (N,).
average_mode (str) – The type of averaging performed on the result. Options are ‘macro’ and ‘none’. If ‘none’, the scores for each class are returned. If ‘macro’, calculate metrics for each class, and find their unweighted mean. Defaults to ‘macro’.
thrs (Number | tuple[Number], optional) – Predictions with scores under the thresholds are considered negative. Default to 0.

返回

Precision.

返回类型

float | np.array | list[float | np.array]

Args	`thrs` is number	`thrs` is tuple
`average_mode` = “macro”	float	list[float]
`average_mode` = “none”	np.array	list[np.array]