forked from mindspore-Ecosystem/mindspore
50 lines
1.7 KiB
ReStructuredText
50 lines
1.7 KiB
ReStructuredText
|
mindspore.nn.ConfusionMatrix
|
|||
|
|
============================
|
|||
|
|
|
|||
|
|
.. py:class:: mindspore.nn.ConfusionMatrix(num_classes, normalize='no_norm', threshold=0.5)
|
|||
|
|
|
|||
|
|
计算混淆矩阵(confusion matrix)，通常用于评估分类模型的性能，包括二分类和多分类场景。
|
|||
|
|
|
|||
|
|
如果您只想使用混淆矩阵，请使用该类。如果想计算"PPV"、"TPR"、"TNR"等，请使用'mindspore.nn.ConfusionMatrixMetric'类。
|
|||
|
|
|
|||
|
|
**参数：**
|
|||
|
|
|
|||
|
|
- **num_classes** (int) - 数据集中的类别数量。
|
|||
|
|
- **normalize** (str) - 计算ConfsMatrix的参数支持四种归一化模式，默认值：None。
|
|||
|
|
|
|||
|
|
- **"no_norm"** (None)：不使用标准化。
|
|||
|
|
- **"target"** (str)：基于目标值的标准化。
|
|||
|
|
- **"prediction"** (str)：基于预测值的标准化。
|
|||
|
|
- **"all"** (str)：整个矩阵的标准化。
|
|||
|
|
|
|||
|
|
- **threshold** (float) - 阈值，用于与输入Tensor进行比较。默认值：0.5。
|
|||
|
|
|
|||
|
|
.. py:method:: clear()
|
|||
|
|
|
|||
|
|
重置评估结果。
|
|||
|
|
|
|||
|
|
.. py:method:: eval()
|
|||
|
|
|
|||
|
|
计算混淆矩阵。
|
|||
|
|
|
|||
|
|
**返回：**
|
|||
|
|
|
|||
|
|
numpy.ndarray，计算的结果。
|
|||
|
|
|
|||
|
|
**异常：**
|
|||
|
|
|
|||
|
|
- **RuntimeError** - 没有先调用update方法。
|
|||
|
|
|
|||
|
|
.. py:method:: update(*inputs)
|
|||
|
|
|
|||
|
|
使用y_pred和y更新内部评估结果。
|
|||
|
|
|
|||
|
|
**参数：**
|
|||
|
|
|
|||
|
|
- ***inputs** (tuple) - 输入 `y_pred` 和 `y` 。 `y_pred` 和 `y` 是 `Tensor` 、列表或数组。
|
|||
|
|
`y_pred` 是预测值， `y` 是真实值， `y_pred` 的shape是 :math:`(N, C, ...)` 或 :math:`(N, ...)` ， `y` 的shape是 :math:`(N, ...)` 。
|
|||
|
|
|
|||
|
|
**异常：**
|
|||
|
|
|
|||
|
|
- **RuntimeError** - 没有先调用update方法。
|
|||
|
|
|