我使用下面的代码片段来使用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'])
结果如下:在这里输入图像描述
发布于 2022-01-10 00:34:30
尝试在代码disp.plot(xticks_rotation = 'vertical')
中添加以下内容
默认情况下,它是水平显示的,但可以更改此行为。
您可以在正式文档https://scikit-learn.org/stable/modules/generated/sklearn.metrics.ConfusionMatrixDisplay.html#sklearn.metrics.ConfusionMatrixDisplay中找到更多详细信息。
https://stackoverflow.com/questions/70594734
复制相似问题