Sklearn混淆矩阵：如何获得蓝色

Posted 2023-03-12

【中文标题】Sklearn混淆矩阵：如何获得蓝色

【英文标题】：Sklearn Confusion Matrix: How to get the blue color

【发布时间】：2021-10-25 02:45:59

【问题描述】：

当我使用此代码绘制我的混淆矩阵时

from sklearn.metrics import confusion_matrix, ConfusionMatrixDisplay
cm = confusion_matrix(y_test, rmc_pred, labels=rmc.classes_)

disp = ConfusionMatrixDisplay(confusion_matrix=cm,
                                display_labels=rmc.classes_)


disp.plot() 

我得到一个紫色、黄色、蓝色和绿色的矩阵，每行都有垂直和水平线切割。我想得到一个看起来像 sklearn 文档中的蓝色混淆矩阵。你们中的任何人都可以帮助我如何做到这一点吗？ 非常感谢！

如果我将 sklearns 函数 plot_confusion_matrix 与 cmap 参数一起使用，我会收到一条错误消息，指出：FixedLocator 位置的数量 (9)，通常来自对 set_ticks 的调用，与刻度标签的数量 (10) 不匹配。

【问题讨论】：

希望答案是有帮助的。彻底回答问题很耗时。如果您的问题已解决，请接受解决方案。 ✔ 位于答案左上角的 ▲/▼ 箭头下方。如果出现更好的解决方案，则可以接受新的解决方案。如果您的声望超过 15，您还可以使用 ▲/▼ 箭头对答案的有用性进行投票。 如果解决方案无法回答问题，请发表评论。 What should I do when someone answers my question?。谢谢。

【参考方案1】：

您可以简单地更改用于显示混淆矩阵的cmap，如下所示：

import matplotlib.pyplot as plt
disp.plot(cmap=plt.cm.Blues)