python-3.x - sklearn.metrics.ConfusionMatrixDisplay 使用科学记数法

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) 显示。我想显示所有数字，并在 ConfusionMatrixDisplay documentation 中找到了 values_format 参数.我试过这样使用它:

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

但是我得到一个类型错误:

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

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

Answer 1

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

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