sklearn.metrics.ConfusionMatrixDisplay¶
class sklearn.metrics.ConfusionMatrixDisplay(confusion_matrix, *, display_labels=None)
混淆矩阵可视化。
建议使用plot_confusion_matrix创建ConfusionMatrixDisplay。 所有参数都存储为属性。
在用户指南中阅读更多内容。
| 参数 | 说明 |
|---|---|
| confusion_matrix | ndarray of shape (n_classes, n_classes) 混淆矩阵 |
| display_labels | ndarray of shape (n_classes,), default=None 显示绘图标签。如果为None，则将显示标签设置为0到n_classes-1。 |
| 属性 | 说明 |
|---|---|
| im_ | matplotlib AxesImage 表示混淆矩阵的图像。 |
| text_ | ndarray of shape (n_classes, n_classes), dtype=matplotlib Text, or None matplotlib轴的数组。如果include_values为false，则为None。 |
| ax_ | matplotlib Axes 带有混淆矩阵的轴。 |
| figure_ | matplotlib Figure 该图包含混淆矩阵。 |
| 方法 | 说明 |
|---|---|
plot(*[, include_values, cmap, …]) |
绘图可视化。 |
__init__(confusion_matrix, *, display_labels=None)
自我初始化。有关精确签名，请参见help（type（self））。
plot(*, include_values=True, cmap='viridis', xticks_rotation='horizontal', values_format=None, ax=None)
绘图可视化。
| 参数 | 说明 |
|---|---|
| include_values | bool, default=True 在混淆矩阵中包含值。 |
| cmap | str or matplotlib Colormap, default=’viridis’ matplotlib识别的颜色图。 |
| xticks_rotation | {‘vertical’, ‘horizontal’} or float, default=’horizontal’ xtick标签的旋转。 |
| values_format | str, default=None 混淆矩阵中值的格式规范。如果为None，则格式说明为‘d’或‘.2g’，以较短者为准。 |
| ax | matplotlib axes, default=None 要绘制的轴对象。如果为None，则将创建新的图形和轴。 |
| 返回值 | 说明 |
|---|---|
| display | ConfusionMatrixDisplay |
