8

在 Matlab 中执行 SVM 训练后,如何可视化 SVM 分类?

到目前为止,我只训练了 SVM:

% Labels are -1 or 1
groundTruth = Ytrain;
d = xtrain;

model = svmtrain(groundTruth, d);
4

4 回答 4

7

如果您使用的是 LIBSVM,则可以绘制分类结果:

% Labels are -1 or 1
groundTruth = Ytrain;
d = xtrain;

figure

% plot training data
hold on;
pos = find(groundTruth==1);
scatter(d(pos,1), d(pos,2), 'r')
pos = find(groundTruth==-1);
scatter(d(pos,1), d(pos,2), 'b')

% now plot support vectors
hold on;
sv = full(model.SVs);
plot(sv(:,1),sv(:,2),'ko');

% now plot decision area
[xi,yi] = meshgrid([min(d(:,1)):0.01:max(d(:,1))],[min(d(:,2)):0.01:max(d(:,2))]);
dd = [xi(:),yi(:)];
tic;[predicted_label, accuracy, decision_values] = svmpredict(zeros(size(dd,1),1), dd, model);toc
pos = find(predicted_label==1);
hold on;
redcolor = [1 0.8 0.8];
bluecolor = [0.8 0.8 1];
h1 = plot(dd(pos,1),dd(pos,2),'s','color',redcolor,'MarkerSize',10,'MarkerEdgeColor',redcolor,'MarkerFaceColor',redcolor);
pos = find(predicted_label==-1);
hold on;
h2 = plot(dd(pos,1),dd(pos,2),'s','color',bluecolor,'MarkerSize',10,'MarkerEdgeColor',bluecolor,'MarkerFaceColor',bluecolor);
uistack(h1, 'bottom');
uistack(h2, 'bottom');
于 2011-02-03T19:16:36.230 回答
2

假设您的数据有两个以上的维度,您可以执行 PCA,将数据投影到 2D,然后根据 svm 分类器的输出为它们分配颜色(例如,A 类为红色,B 类为蓝色)。这很快就可以做到,您将看到是否有任何可视化的东西。但是,具有高维数的数据不太可能在 2D 中轻松可视化。

于 2011-02-03T21:22:43.560 回答
0

查看这个svm-toy函数,类似于 LIBSVM 中的函数。显然它只适用于二维二进制分类

于 2010-04-16T00:32:04.497 回答
-1
model = svmtrain(groundTruth, d, 'ShowPlot', true);
于 2014-10-31T20:26:58.593 回答