gpt4 book ai didi

python - Sklearn 混淆矩阵 : How to get the blue color

转载 作者:行者123 更新时间:2023-12-05 01:28:10 31 4
gpt4 key购买 nike

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

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) 不匹配。

最佳答案

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

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

关于python - Sklearn 混淆矩阵 : How to get the blue color,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/68914814/

31 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com