调整ConfusionMatrixDisplay (ScikitLearn)的大小
调整ConfusionMatrixDisplay (ScikitLearn)的大小
提问于 2021-03-04 21:36:49
如何设置由山猫学习的混淆矩阵绘制的数字的大小？
import numpy as np
from sklearn.metrics import ConfusionMatrixDisplay, confusion_matrix
cm = confusion_matrix(np.arange(25), np.arange(25))
cmp = ConfusionMatrixDisplay(cm, display_labels=np.arange(25))
cmp.plot()上面的代码显示了这个图，它太紧了：
回答 2
Stack Overflow用户
回答已采纳
发布于 2021-03-04 21:47:20
可以将matplotlib.axes对象发送到sklearn.metrics.ConfusionMatrixDisplay的.plot方法。首先在matplotlib.pyplot.subplots中设置图形的大小。
import numpy as np
from sklearn.metrics import ConfusionMatrixDisplay, confusion_matrix
import matplotlib.pyplot as plt
cm = confusion_matrix(np.arange(25), np.arange(25))
cmp = ConfusionMatrixDisplay(cm, display_labels=np.arange(25))
fig, ax = plt.subplots(figsize=(10,10))
cmp.plot(ax=ax)
Stack Overflow用户
发布于 2022-10-14 06:00:49
我正在寻找如何调整颜色栏，就像@Raphael提供的答案中有人在评论中指出的那样，现在我想添加如何做这个。
我使用了ConfusionMatrixDisplay的属性，并在这个answer的指导下修改了代码，以便：
import numpy as np
from sklearn.metrics import ConfusionMatrixDisplay, confusion_matrix
import matplotlib.pyplot as plt
cm = confusion_matrix(np.arange(25), np.arange(25))
cmp = ConfusionMatrixDisplay(cm, display_labels=np.arange(25))
fig, ax = plt.subplots(figsize=(10,10))
# Deactivate default colorbar
cmp.plot(ax=ax, colorbar=False)
# Adding custom colorbar
cax = fig.add_axes([ax.get_position().x1+0.01,ax.get_position().y0,0.02,ax.get_position().height])
plt.colorbar(cmp.im_, cax=cax)
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/66483409
复制相关文章
相似问题
腾讯云开发者
Copyright © 2013 - 2026 Tencent Cloud. All Rights Reserved. 腾讯云 版权所有
深圳市腾讯计算机系统有限公司 ICP备案/许可证号：粤B2-20090059 
粤公网安备44030502008569号
腾讯云计算（北京）有限责任公司 京ICP证150476号 | 京ICP备11018762号