sklearn.metrics.ConfusionMatrixDisplay 使用科学记数法

sklearn.metrics.ConfusionMatrixDisplay using scientific notation

我正在生成一个混淆矩阵,如下所示:

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'.

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

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

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