【问题标题】:sklearn.metrics.ConfusionMatrixDisplay using scientific notationsklearn.metrics.ConfusionMatrixDisplay 使用科学计数法
【发布时间】:2021-04-04 09:15:07
【问题描述】:

我正在生成如下混淆矩阵:

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

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

【问题讨论】:

    标签: python-3.x scikit-learn confusion-matrix


    【解决方案1】:

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

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

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 2011-07-18
      • 1970-01-01
      • 2016-05-27
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多