备注
您正在阅读 MMClassification 0.x 版本的文档。MMClassification 0.x 会在 2022 年末被切换为次要分支。建议您升级到 MMClassification 1.0 版本，体验更多新特性和新功能。请查阅 MMClassification 1.0 的安装教程、迁移教程以及更新日志。
mmcls.core.calculate_confusion_matrix¶
- mmcls.core.calculate_confusion_matrix(pred, target)[源代码]¶
Calculate confusion matrix 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,).
- 返回
- Confusion matrix
The shape is (C, C), where C is the number of classes.
- 返回类型