matlab fitensemble 绘图 roc 误差

Question

我正在使用 adaboost 构建一个模型，并尝试让 roc 图发挥作用。这是我的代码:

ens=fitensemble(X,y,'AdaBoostM1',100,'Tree');

[ytest, scores] = predict(ens,Xtest);

figure
[xx,yy] = perfcurve(label, scores(:,2),'yes');
plot(xx,yy)
xlabel('FPR')
ylabel('TPR')
title('ROC');

但是，这给了我一个错误:

Error using perfcurve>membership (line 693)
Positive class is not found in the input data.

我的训练数据大小是 1000x19，测试数据大小是 100x19。

这是来自 matlab 的源代码: https://www.mathworks.com/matlabcentral/fileexchange/42744-machine-learning-with-matlab?focused=6797233&tab=example

Answer 1

解决了。必须将标签格式从逻辑格式更改为数字格式，因为我的标签是 0 和 1。

这是更正后的代码:

ens=fitensemble(X,y,'AdaBoostM1',100,'Tree');

[ytest, scores] = predict(ens,Xtest);

figure
[xx,yy] = perfcurve(label, scores(:,2),1);
plot(xx,yy)
xlabel('FPR')
ylabel('TPR')
title('ROC');