Sklearn 混淆矩阵：更改颜色条

Question

我认为答案应该非常简单，但我找不到立即解决问题的方法。我有一个混淆矩阵，我想将颜色更改为蓝色（等级类型）。

是否可以使用“colobar”以简单的方式做到这一点？

plot_confusion_matrix(clf_LR9, X1_test, y1_test)

提前致谢

Answer 1

您应该将 cmap 参数设置为 "Blues"。

plot_confusion_matrix(clf_LR9, X1_test, y1_test, cmap="Blues")