通过pandas中的crosstab调用混淆矩阵,来查看哪些结果最容易混淆,对角线部分,为正确预测的部分,其它部分则是预测错误的部分。
import pandas as pd
pd.crosstab(test_labels,prediction,rownames=['label'],colnames=['predict'])
查看预测错误位置的数据:
df = pd.DataFrame({'label':test_labels,'predict':prediction})
print(df[(df.label==5)&(df.predict==3)]) #查看是5预测错成3的数据位置