1 tic %计时器2 %%清空环境变量3 close all4 clear5 clc6 format compact7 %%数据提取8 % 载入测试数据wine,其中包含的数据为classnumber = 3,wine:178*13的矩阵,wine_labes:178*1的列向量9 load wine.mat10 %选定训练集和测试集11 % 将第一类的1-30,第二类的60-95,第三类的131-153做为训练集12 train_wine = [wine(1:30,:);wine(60:95,:);wine(131:153,:)];13 %相应的训练集的标签也要分离出来14 train_wine_labels = [wine_labels(1:30);wine_labels(60:95);wine_labels(131:153)];15 % 将第一类的31-59,第二类的96-130,第三类的154-178做为测试集16 test_wine = [wine(31:59,:);wine(96:130,:);wine(154:178,:)];17 %相应的测试集的标签也要分离出来18 test_wine_labels = [wine_labels(31:59);wine_labels(96:130);wine_labels(154:178)];19 %%数据预处理20 % 数据预处理,将训练集和测试集归一化到[0,1]区间21 [mtrain,ntrain] =size(train_wine);22 [mtest,ntest] =size(test_wine);23
24 dataset =[train_wine;test_wine];25 %mapminmax为MATLAB自带的归一化函数26 [dataset_scale,ps] = mapminmax(dataset',0,1);
27 dataset_scale = dataset_scale';
28
29 train_wine = dataset_scale(1:mtrain,:);30 test_wine = dataset_scale( (mtrain+1):(mtrain+mtest),: );31 %%利用灰狼算法选择最佳的SVM参数c和g32 SearchAgents_no=10; %狼群数量,Number of search agents33 Max_iteration=10; %最大迭代次数,Maximum numbef of iterations34 dim=2; %此例需要优化两个参数c和g,number of your variables35 lb=[0.01,0.01]; %参数取值下界36 ub=[100,100]; %参数取值上界37 % v = 5; %SVM Cross Validation参数,默认为538
39 %initialize alpha, beta, and delta_pos40 Alpha_pos=zeros(1,dim); %初始化Alpha狼的位置41 Alpha_score=inf; % 初始化Alpha狼的目标函数值,change this to -inf formaximization problems42
43 Beta_pos=zeros(1,dim); %初始化Beta狼的位置44 Beta_score=inf; % 初始化Beta狼的目标函数值,change this to -inf formaximization problems45
46 Delta_pos=zeros(1,dim); %初始化Delta狼的位置47 Delta_score=inf; % 初始化Delta狼的目标函数值,change this to -inf formaximization problems48
49 %Initialize the positions of search agents50 Positions=initialization(SearchAgents_no,dim,ub,lb);51
52 Convergence_curve=zeros(1,Max_iteration);53
54 l=0; %Loop counter循环计数器55
56 %Main loop主循环57 while l
60 %Return back the search agents that go beyond the boundaries of the search space61 %若搜索位置超过了搜索空间,需要重新回到搜索空间62 Flag4ub=Positions(i,:)>ub;63 Flag4lb=Positions(i,:)
68 %计算适应度函数值69 cmd = ['-c',num2str(Positions(i,1)),'-g',num2str(Positions(i,2))];70 model=svmtrain(train_wine_labels,train_wine,cmd); %SVM模型训练71 [~,fitness]=svmpredict(test_wine_labels,test_wine,model); %SVM模型预测及其精度72 fitness=100-fitness(1); %以错误率最小化为目标73
74 %Update Alpha, Beta, and Delta75 if fitness
80 if fitness>Alpha_score && fitness
85 if fitness>Alpha_score && fitness>Beta_score && fitness
91 a=2-l*((2)/Max_iteration); % 对每一次迭代,计算相应的a值,a decreases linearly fron 2 to 0
92
93 %Update the Position of search agents including omegas94 for i=1:size(Positions,1) %遍历每个狼95 for j=1:size(Positions,2) %遍历每个维度96
97 %包围猎物,位置更新98
99 r1=rand(); % r1 is a random number in [0,1]100 r2=rand(); % r2 is a random number in [0,1]101
102 A1=2*a*r1-a; % 计算系数A,Equation (3.3)103 C1=2*r2; % 计算系数C,Equation (3.4)104
105 %Alpha狼位置更新106 D_alpha=abs(C1*Alpha_pos(j)-Positions(i,j)); % Equation (3.5)-part 1
107 X1=Alpha_pos(j)-A1*D_alpha; % Equation (3.6)-part 1
108
109 r1=rand();110 r2=rand();111
112 A2=2*a*r1-a; % 计算系数A,Equation (3.3)113 C2=2*r2; % 计算系数C,Equation (3.4)114
115 %Beta狼位置更新116 D_beta=abs(C2*Beta_pos(j)-Positions(i,j)); % Equation (3.5)-part 2
117 X2=Beta_pos(j)-A2*D_beta; % Equation (3.6)-part 2
118
119 r1=rand();120 r2=rand();121
122 A3=2*a*r1-a; % 计算系数A,Equation (3.3)123 C3=2*r2; % 计算系数C,Equation (3.4)124
125 %Delta狼位置更新126 D_delta=abs(C3*Delta_pos(j)-Positions(i,j)); % Equation (3.5)-part 3
127 X3=Delta_pos(j)-A3*D_delta; % Equation (3.5)-part 3
128
129 %位置更新130 Positions(i,j)=(X1+X2+X3)/3;% Equation (3.7)131
132 end133 end134 l=l+1;135 Convergence_curve(l)=Alpha_score;136 end137 bestc=Alpha_pos(1,1);138 bestg=Alpha_pos(1,2);139 bestGWOaccuarcy=Alpha_score;140 %%打印参数选择结果141 disp('打印选择结果');142 str=sprintf('Best Cross Validation Accuracy = %g%%,Best c = %g,Best g = %g',bestGWOaccuarcy*100,bestc,bestg);143 disp(str)144 %%利用最佳的参数进行SVM网络训练145 cmd_gwosvm = ['-c',num2str(bestc),'-g',num2str(bestg)];146 model_gwosvm =svmtrain(train_wine_labels,train_wine,cmd_gwosvm);147 %%SVM网络预测148 [predict_label,accuracy] =svmpredict(test_wine_labels,test_wine,model_gwosvm);149 %打印测试集分类准确率150 total =length(test_wine_labels);151 right = sum(predict_label ==test_wine_labels);152 disp('打印测试集分类准确率');153 str = sprintf( 'Accuracy = %g%% (%d/%d)',accuracy(1),right,total);154 disp(str);155 %%结果分析156 %测试集的实际分类和预测分类图157 figure;158 hold on;159 plot(test_wine_labels,'o');160 plot(predict_label,'r*');161 xlabel('测试集样本','FontSize',12);162 ylabel('类别标签','FontSize',12);163 legend('实际测试集分类','预测测试集分类');164 title('测试集的实际分类和预测分类图','FontSize',12);165 grid on166 snapnow167 %%显示程序运行时间168 toc