具有ROC分析的KNN分类器

时间:2013-03-19 12:34:21

标签: matlab roc

我写了以下代码来获取我的KNN分类器的ROC图:

load fisheriris;

features                                = meas;
featureSelcted                          = features;
numFeatures                             = size(meas,1);

%% Define ground truth
groundTruthGroup                        = species;

%% Construct a KNN classifier
KNNClassifierObject                     = ClassificationKNN.fit(featureSelcted, groundTruthGroup, 'NumNeighbors', 3, 'Distance', 'euclidean');

% Predict resubstitution response of k-nearest neighbor classifier
[KNNLabel, KNNScore]                    = resubPredict(KNNClassifierObject);

% Fit probabilities for scores
groundTruthNumericalLable           = [ones(50,1); zeros(50,1); -1.*ones(50,1)];
[FPR, TPR, Thr, AUC, OPTROCPT]          = perfcurve(groundTruthNumericalLable(:,1), KNNScore(:,1), 1);

然后我们可以绘制FPR与TPR以获得ROC曲线。

然而,FPR和TPR与我使用我自己的实现不同,上面的那个不会显示所有的点,实际上,上面的代码只显示ROC上的三个点。我实施的代码将在ROC上显示151个点,因为数据的大小为150。

patternsKNN                             = [KNNScore(:,1), groundTruthNumericalLable(:,1)];
patternsKNN                             = sortrows(patternsKNN, -1);
groundTruthPattern                      = patternsKNN(:,2);

POS                                     = cumsum(groundTruthPattern==1);
TPR                                     = POS/sum(groundTruthPattern==1);
NEG                                     = cumsum(groundTruthPattern==0);
FPR                                     = NEG/sum(groundTruthPattern==0);

FPR                                     = [0; FPR];
TPR                                     = [0; TPR];

请问如何调整' perfcurve'让它输出ROC的所有积分?非常感谢。

一个。

1 个答案:

答案 0 :(得分:1)

我对perfcurve不熟悉,但您是否尝试过使用MALAB帮助。 在MATLAB工作区中输入:help perfcurve,你会得到很多帮助。 同样在MATWORKS网站上还有帮助页面,例如你的案例看看: http://www.mathworks.com/help/stats/perfcurve.html