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

Question

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

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 参数一起使用，我会收到一条错误消息，指出：固定定位器位置的数量 (9)，通常来自对 set_ticks 的调用，与刻度标签的数量 (10) 不匹配。

Answer 1

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

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