Multi-Class SVM( one versus all)

前端 未结 3 406
粉色の甜心
粉色の甜心 2020-12-08 09:07

I know that LIBSVM only allows one-vs-one classification when it comes to multi-class SVM. However, I would like to tweak it a bit to perform one-against-all classification.

3条回答
  •  孤城傲影
    2020-12-08 09:21

    %# Fisher Iris dataset
    load fisheriris
    [~,~,labels] = unique(species);   %# labels: 1/2/3
    data = zscore(meas);              %# scale features
    numInst = size(data,1);
    numLabels = max(labels);
    
    %# split training/testing
    idx = randperm(numInst);
    numTrain = 100; numTest = numInst - numTrain;
    trainData = data(idx(1:numTrain),:);  testData = data(idx(numTrain+1:end),:);
    trainLabel = labels(idx(1:numTrain)); testLabel = labels(idx(numTrain+1:end));
    %# train one-against-all models
    model = cell(numLabels,1);
    for k=1:numLabels
        model{k} = svmtrain(double(trainLabel==k), trainData, '-c 1 -g 0.2 -b 1');
    end
    
    %# get probability estimates of test instances using each model
    prob = zeros(numTest,numLabels);
    for k=1:numLabels
        [~,~,p] = svmpredict(double(testLabel==k), testData, model{k}, '-b 1');
        prob(:,k) = p(:,model{k}.Label==1);    %# probability of class==k
    end
    
    %# predict the class with the highest probability
    [~,pred] = max(prob,[],2);
    acc = sum(pred == testLabel) ./ numel(testLabel)    %# accuracy
    C = confusionmat(testLabel, pred)                   %# confusion matrix
    

提交回复
热议问题