sklearn.metrics.ConfusionMatrix使用科学计数法显示

Question

我正在生成一个混淆矩阵，如下所示：

from sklearn.metrics import confusion_matrix, ConfusionMatrixDisplay
cm = confusion_matrix(truth_labels, predicted_labels, labels=n_classes)
disp = ConfusionMatrixDisplay(confusion_matrix=cm)
disp = disp.plot(cmap="Blues")
plt.show()

然而，我的一些真阳性、真阴性等值超过 30,000，并且它们以科学格式 (3e+04) 显示。我想显示所有数字并values_format在 ConfusionMatrixDisplay文档中找到了该参数。我尝试过像这样使用它：

disp = ConfusionMatrixDisplay(confusion_matrix=cm, values_format='')

但我收到类型错误：

TypeError: __init__() got an unexpected keyword argument 'values_format'.

我做错了什么？提前致谢！

Answer 1

如果有人遇到同样的问题，我刚刚找到了答案。value_format 参数必须添加到 disp.plot 中，而不是添加到 ConfusionMatrixDisplay 调用中，如下所示：

disp.plot(cmap="Blues", values_format='')