【问题标题】:How to write horizontal label properly in confusion matrix?如何在混淆矩阵中正确书写水平标签?
【发布时间】:2022-01-10 20:53:41
【问题描述】:

我正在使用以下代码 sn-p 使用 sklearn lib 绘制混淆矩阵。

from sklearn.metrics import confusion_matrix,ConfusionMatrixDisplay   
cm=confusion_matrix(y_test,y_pred,normalize='true')
        
disp=ConfusionMatrixDisplay(confusion_matrix=cm,display_labels=['anger','bordome','disgust','fear', 'happiness','sadness' ,'neutral'])

结果如下: enter image description here

【问题讨论】:

    标签: scikit-learn confusion-matrix


    【解决方案1】:

    尝试在您的代码中添加它 disp.plot(xticks_rotation = 'vertical')

    默认情况下,它是水平显示的,但可以更改此行为。

    您可以在官方文档中找到更多详细信息 https://scikit-learn.org/stable/modules/generated/sklearn.metrics.ConfusionMatrixDisplay.html#sklearn.metrics.ConfusionMatrixDisplay

    【讨论】:

      猜你喜欢
      • 2020-08-30
      • 1970-01-01
      • 2020-01-22
      • 2021-10-07
      • 2020-07-31
      • 2018-11-06
      • 1970-01-01
      • 1970-01-01
      • 2011-01-10
      相关资源
      最近更新 更多