Seaborn并没有显示混淆矩阵单元格中的所有数字

2024-04-24 00:43:09 发布

您现在位置:Python中文网/ 问答频道 /正文

我用来绘制混淆矩阵的函数是:

def plotConfMatrix(title,cm):
    sb.set(font_scale=1.4)
    cm_df = pd.DataFrame(cm,columns= cm.keys(),index=cm.keys())
    #cm_df.index.name = 'Predicted'
    #cm_df.columns.name = 'Actual'
    sb.heatmap(cm_df, annot=True,annot_kws={"size": 16},fmt='d')
    plt.xlabel('Actual labels')
    plt.ylabel('Predicted labels') 
    plt.title('Confusion Matrix for ' + title)
    plt.show()

在此函数中,混淆矩阵作为dict对象输入,并转换为dataframe。但它只显示了其中一个细胞的价值。我试着升级seaborn版本,但没有成功。Dict对象是:

^{pr2}$

它转换成的数据帧是:

    0   1
0  36   6
1   9  29

情节是:

Confusion matrix

可能的问题是什么?另外,我使用Python空闲编辑器和Ubuntu终端中的普通pythonshell来执行这段代码。在


Tags: columns函数namedfindexlabelstitlecm