【有问不答】混淆矩阵可视化（MATLAB）

今天不飞了

效果像这样

直接上代码咯

---

代码

% 这是你已经统计好的
TN = 154200;
FP = 1973;
FN = 11036;
TP = 145137;

% 生成数量矩阵和概率矩阵
ConfMat = [TN,FP;FN,TP];
ProbMat = round(ConfMat/sum(ConfMat(:))*100,2);

% 绘制咯
imagesc(ConfMat),colorbar
colormap(mycolor) % 务必换成你自己的map，这是我自己做的，你没有。运行会报错的哈

% 设置坐标
xticks([1,2])
xticklabels({'0','1'})
yticks([1,2])
yticklabels({'0','1'})

% 填写信息
T = 30;
g = 0.2;
text(1-g,1,{'True Neg';num2str(ConfMat(1));strcat(num2str(ProbMat(1)),'%')},...
    'Color',[1,1,1]*double(ProbMat(1)>T))
text(1-g,2,{'False Pos';num2str(ConfMat(2));strcat(num2str(ProbMat(2)),'%')},...
    'Color',[1,1,1]*double(ProbMat(2)>T))
text(2-g,1,{'False Neg';num2str(ConfMat(3));strcat(num2str(ProbMat(3)),'%')},...
    'Color',[1,1,1]*double(ProbMat(3)>T))
text(2-g,2,{'True Pos';num2str(ConfMat(4));strcat(num2str(ProbMat(4)),'%')},...
    'Color',[1,1,1]*double(ProbMat(4)>T))

这里有个骚操作，就是在写文本的时候加入了'Color',[1,1,1]*double(ProbMat(4)>T)，它的作用就是控制字体的颜色，背景深则显示亮色，反之显示深色。 6666

---

其他

1. 你有其他问题吗，欢迎留言告诉我