本算法根据《机器学习实战》中Python版的完整版SMO改编而来,所以详细的过程说明请参照原书。由于代码量较大,函数较多,所以将整个算法和数据进行了打包,可在云盘上下载链接: https://pan.baidu.com/s/1eS6g0s6 密码: pat5。这里只给出主程序的代码及实验结果。需要注意的是,因为实验具有随机性,每次仿真的结果可能会有一定的差异。
以下为实验Matlab代码:
clc;
clear;
%加载测试数据文件,前两列为坐标值,后两列为类标号
fileID = fopen('D:\matlabFile\SVM\SVM.txt');
DS=textscan(fileID,'%f %f %f');
fclose(fileID);
%将数据转为矩阵形式
DataMat=cat(2,DS{1},DS{2});
[Row,Column]=size(DataMat);
%提取类别矩阵
LabelMat=DS{3};
%常数
C=0.6;
%容差
Toler=0.001;
%设定最大循环次数
MaxIter=40;
kTup={'lin',0};
%构造结构体
OS=BuildStruct(DataMat,LabelMat,C,Toler,kTup);
%循环计数器
flag=0;
EnterSet=true;
alphaPairChanged=0;
while (flag<MaxIter)&&(alphaPairChanged>0||EnterSet)
alphaPairChanged=0;
if EnterSet
for i=1:OS.m
Result=InnerL(i,OS);
OS=Result{1,1};
alphaPairChanged=alphaPairChanged+Result{1,2};
end
flag=flag+1;
else
Index=find(OS.Alpha>0);
for i=1:length(Index)
Result=InnerL(Index(i),OS);
OS=Result{1,1};
alphaPairChanged=alphaPairChanged+Result{1,2};
end
flag=flag+1;
end
if EnterSet
EnterSet=false;
elseif alphaPairChanged==0
EnterSet=true;
end
end
%显示原始点集
scatter(DataMat(:,1),DataMat(:,2),'filled');
hold on
%显示支持向量
Index=find(OS.Alpha>0);
Sup=DataMat(Index,:);
scatter(Sup(:,1),Sup(:,2),'r','filled');
以下为实验结果图:

版权声明:本文为博主原创文章,未经博主允许不得转载。