我们通常使用ROC曲线来评价分类结果的好坏,在MATLAB中绘制该曲线其实也十分容易。我们让label表示真实的类别,output表示预测的类别,那么调用:
[XRF,YRF,TRF,AUCRF] = perfcurve(label,output,1);
之后运行:
plot(XRF,YRF)
即可得到ROC曲线,该函数中还有一个参数,也就是正类的标签(positive class label),该变量是一个常数,可以为数值,也可是是字符串等,需要指定一下。上面的例子中,我们输出的结果和标签都是0或1的数值型向量,因此指定1为正类。AUCRF即得到我们通常所说的AUC的值。