如何在混淆矩阵中正确书写水平标签?
How to write horizontal label properly in confusion matrix?
我正在使用以下代码片段使用 sklearn 库绘制混淆矩阵。
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
尝试将此添加到您的代码中
disp.plot(xticks_rotation = 'vertical')
默认情况下,它是水平显示的,但可以更改此行为。
您可以在官方文档中找到更多详细信息
https://scikit-learn.org/stable/modules/generated/sklearn.metrics.ConfusionMatrixDisplay.html#sklearn.metrics.ConfusionMatrixDisplay
我正在使用以下代码片段使用 sklearn 库绘制混淆矩阵。
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
尝试将此添加到您的代码中
disp.plot(xticks_rotation = 'vertical')
默认情况下,它是水平显示的,但可以更改此行为。
您可以在官方文档中找到更多详细信息 https://scikit-learn.org/stable/modules/generated/sklearn.metrics.ConfusionMatrixDisplay.html#sklearn.metrics.ConfusionMatrixDisplay