matplotlib画混淆矩阵和正确率曲线

最新推荐文章于 2024-08-19 20:05:05 发布

小鱼爱吃肉

最新推荐文章于 2024-08-19 20:05:05 发布

阅读量3.1k

点赞数 2

本文链接：https://blog.csdn.net/yuan0401yu/article/details/88730555

版权

本文介绍了如何利用matplotlib库在Python中绘制混淆矩阵和正确率曲线。虽然找不到原先参考的博客，但作者分享了自己绘制这些图表的经验。

摘要由CSDN通过智能技术生成

今天很兴奋，再来一篇庆祝一下~~~

混淆矩阵

找不到参看的那篇博客啦~~希望原博主不要讨伐我

#!/usr/bin/python3.5
# -*- coding: utf-8 -*-
import numpy as np
import matplotlib.pyplot as plt
plt.rcParams['font.sans-serif'] = ['FangSong']  #可显示中文字符
plt.rcParams['axes.unicode_minus']=False

classes = ['a','b','c','d','e','f','g']
confusion_matrix = np.array([(99,1,2,2,0,0,6),(1,98,7,6,2,1,1),(0,0,86,0,0,2,0),(0,0,0,86,1,0,0),(0,0,0,1,94,1,0),(0,1,5,1,0,96,8),(0,0,0,4,3,0,85)],dtype=np.float64)

plt.imshow(confusion_matrix, interpolation='nearest', cmap=plt.cm.Oranges)  #按照像素显示出矩阵
plt.title('混淆矩阵')
plt.colorbar()
tick_marks = np.arange(len(classes))
plt.xticks(tick_marks, classes, rotation=-45)
plt.yticks(tick_marks, classes)

thresh = confusion_matrix.max() / 2.
#iters = [[i,j] for i in range(len(classes)) for