% predict - 分类器对测试集的分类结果
% ground_truth - 测试集的正确标签,这里只考虑二分类,即2和4
% auc - 返回ROC曲线的曲线下的面积
function auc = plotroc( decision_values, test_set_labels )
%初始点为(1.0, 1.0)
%计算出ground_truth中正样本的数目pos_num和负样本的数目neg_num
pos_num = sum(test_set_labels==1);
neg_num = sum(test_set_labels==0);
m=size(test_set_labels,1);%返回行数(测试集大小)
[pre,Index]=sort(decision_values);%从低到高排序
test_set_label=test_set_labels(Index);
x=zeros(m+1,1);
y=zeros(m+1,1);
auc=0;
x(1)=1;y(1)=1;%初始点为(1.0, 1.0)
for i=2:m
TP=sum(test_set_label(i:m)==1);FP=sum(test_set_label(i:m)==0);
x(i)=FP/neg_num;
y(i)=TP/pos_num;
auc=auc+(y(i)+y(i-1))*(x(i-1)-x(i))/2;
end;
x(m+1)=0;y(m+1)=0;
auc=auc+y(m)*x(m)/2;
figure;
plot(x,y,'k','linewidth',2);
title(['AUC=',num2str(auc)])%AUC
xlabel('false positive rate')
ylabel('true positive rate')
end
plotroc
最新推荐文章于 2023-03-12 15:39:30 发布