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 |
