精致混淆矩阵和ROC曲线绘制代码分享

基于 MATLAB 代码的创建和展示一个混淆矩阵，并且绘制 ROC 曲线。这两个工具常用于统计分类模型的性能。

1. 混淆矩阵的创建和展示：

   • 数据读取与处理：首先，代码从两个 CSV 文件中读取数据：一个包含了真实的分类结果 （y_true），另一个包含了模型预测的结果 （'y_predy_pred)。
   • 混淆矩阵生成：使用confusionmat函数，代码根据真实类别和预测类别生成混淆矩阵。混淆矩阵是一个表格，显示了每个类别被正确分类和误分类的次数。
   • 归一化：混淆矩阵接着被归一化，这意味着每行的数值被转换为比例，便于比较不同类别的分类性能。
   • 可视化：使用 'imagescimagesc函数，代码将归一化后的混淆矩阵以图像形式展示，其中颜色深浅代表不同的数值。配色方案由ColorMap_me定义。

2. 混淆矩阵的美化与标注：

   • 美化：代码设置了图
   • 标注： 在混淆矩阵的每个格子中，代码添加了显示每个类别的分类准确率的文本。文本的字体、大小和颜色都被精心设置，以确保清晰可见。

3. ROC 曲线绘制：

   • ROC 曲线功能：代码roc_curve 函数（该函数在代码中没有定义，可能是一个自定义函数或来自于某个工具箱），它应该用于生成接收者操作特征（ROC）曲线的数据。ROC 曲线是评估分类模型性能的重要工具，尤其是在处理不平衡数据集时。
   • 功能重要性： ROC 曲线通过描绘真正率（TPR）和假正率（FPR）之间的关系，提供了模型分类阈值的敏感性和特异性的综合视图。

4. 代码

clc
clear
close all
data=readmatrix('真实结果.csv');  y_true=data(:,end);
data=readmatrix('预测结果.csv');  y_pred=data(:,1);
load ColorMap_me
size_num=18;
% 创建混淆矩阵
[confMat,order] = confusionmat(y_true, y_pred);

% 归一化混淆矩阵
confMat = confMat ./ sum(confMat, 2);

% 显示混淆矩阵
fig = figure;
colormap(ColorMap_me); % 使用jet配色方案
% 创建色棒并设置字体为Times New Roman
% 创建色棒并设置字体
cb = colorbar;
cb.FontName = 'Times New Roman';
cb.FontSize = 18;
imagesc(confMat);
title('Confusion Matrix', 'FontName', 'Times New Roman', 'FontSize', size_num);
colorbar;

% 设置轴标签和标题的字体
ax = gca;
ax.XAxis.FontName = 'Times New Roman';
ax.YAxis.FontName = 'Times New Roman';
ax.XAxis.FontSize = size_num; % 调整字号大小
ax.YAxis.FontSize = size_num; % 调整字号大小

% % 设置轴标签的位置
% ax.XAxisLocation = 'top';

% 添加标签
xticks(1:length(order));
yticks(1:length(order));
xticklabels(order);
yticklabels(order);
xlabel('Predicted label', 'FontName', 'Times New Roman', 'FontSize', size_num);
ylabel('True label', 'FontName', 'Times New Roman', 'FontSize', size_num);

% 在每个格子中添加准确率值
numClasses = size(confMat, 1);
textFontSize = size_num; % 字体大小
for i = 1:numClasses
    for j = 1:numClasses
        text(j, i, num2str(confMat(i, j), '%0.2f'), ...
            'HorizontalAlignment', 'center', ...
            'VerticalAlignment', 'middle', ...
            'FontName', 'Times New Roman', ...
            'FontSize', textFontSize, ...
            'Color', 'black');
    end
end

% 调整坐标轴的位置和大小，以减少方框内的空白区域
axis tight;
% 使每个格子都是正方形
axis equal;

roc_data=roc_curve(y_true, y_pred);