修复混淆矩阵图线答案

【问题标题】：Fixing Confusion Matrix plot lines修复混淆矩阵图线
【发布时间】：2021-10-02 03:00:00
【问题描述】：

我正在尝试绘制如下所示的混淆矩阵

cm  = confusion_matrix(testY.argmax(axis=1), predictions.argmax(axis=1))

disp = ConfusionMatrixDisplay(confusion_matrix=cm, display_labels=lb.classes_)
disp = disp.plot(include_values=True, cmap='viridis', ax=None, xticks_rotation='horizontal')

plt.show()

结果：

如您所见，它显示的是框的轴，而不是框的轮廓。由于轴，我看不到黄色框外的数字。我不擅长情节。所以我不知道我需要改变什么。

我的期望：

找到解决方案

plt.tick_params(axis=u'both', which=u'both',length=0)
plt.grid(b=None)

【问题讨论】：

你可以使用plot_confusion_matrix
尝试使用sns plot，它很简单，写在matplotlib之上

标签： python scikit-learn confusion-matrix

【解决方案1】：

关闭网格

例如，

import matplotlib.pyplot as plt
fig, _ = plt.subplots(nrows=1, figsize=(10,10))
ax = plt.subplot(1, 1, 1)
ax.grid(False)

...

disp = ConfusionMatrixDisplay(...)
_ = disp.plot(..., ax=ax, ...)

【讨论】：

【解决方案2】：

cm  = confusion_matrix(testY.argmax(axis=1), predictions.argmax(axis=1))

disp = ConfusionMatrixDisplay(confusion_matrix=cm, display_labels=lb.classes_)
disp = disp.plot(include_values=True, cmap='viridis', ax=None, xticks_rotation='horizontal')
plt.grid(False)
plt.show()

【讨论】：

【解决方案3】：

您作为示例显示的图表是 sns 图。您可以使用 sns 热图来绘制矩阵。

import seaborn as sns
categories = lb.classes_
sns.heatmap(cm, annot=True,categories =categories, cmap='Blues')

【讨论】：

【解决方案4】：

在plot() 函数中更改您的cmap 参数。它代表用颜色对整数值进行颜色映射。

检查

https://matplotlib.org/3.1.0/tutorials/colors/colormaps.html

了解更多详情。

作为答案

cm  = confusion_matrix(testY.argmax(axis=1), predictions.argmax(axis=1))

disp = ConfusionMatrixDisplay(confusion_matrix=cm, display_labels=lb.classes_)
disp = disp.plot(include_values=True, cmap='Blues', ax=None, xticks_rotation='horizontal')

plt.show()

【讨论】：

这会解决颜色问题，但不是他的实际问题，你没有得到实际问题。
他在处理颜色和坐标轴时遇到问题，我给出了颜色问题的解决方案，因为他已经知道如何修复坐标轴了。