因为混淆矩阵的结果是数组类型的,ndarray,所以按照输出二维列表就行。
代码如下:
# 画混淆矩阵
Confuse = confusion_matrix(target_list, pred_list, labels=[0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
output = open('confuse_res20.xls', 'w', encoding='gbk')
output.write('confuse_res20\n')
for i in range(len(Confuse)):
for j in range (len(Confuse[i])):
output.write(str(Confuse[i][j])) # write函数不能写int类型的参数,所以使用str()转化
output.write('\t')
output.write('\n') # 写完一行立马换行
output.close()