找回密码
 注册
关于网站域名变更的通知
查看: 548|回复: 1
打印 上一主题 下一主题

PSO算法特征选择MATLAB实现(单目标)

[复制链接]

该用户从未签到

跳转到指定楼层
1#
发表于 2020-10-20 17:19 | 只看该作者 |只看大图 回帖奖励 |倒序浏览 |阅读模式

EDA365欢迎您登录!

您需要 登录 才可以下载或查看,没有帐号?注册

x
8 O2 K; W5 f7 _) a+ F, _
PSO进行进行特征选择其主要思想是:将子集的选择看作是一个搜索寻优问题(wrapper方法),生成不同的组合,对组合进行评价,再与其他的组合进行比较。这样就将子集的选择看作是一个是一个优化问题。
3 s6 N! M  o, d. G- G5 u
; d9 \9 L3 P5 Z0 q, Q8 B下面是PSO进行特征选择的代码(注意:整体代码是单目标只优化错误率,注意训练使用的是林志仁SVM,数据集是Parkinson,可以到UCI上下载,训练的结果是错误率)9 E- c( A" p/ F5 `& C* A$ w

% I4 l6 L) [$ h; T0 W6 a数据集分割为训练集和测试集:
/ s- W' V3 i- t- y5 `6 r3 R+ y+ }$ J( `# n, [/ a5 V8 F
  • function divide_datasets
  • load Parkinson.mat;
  • dataMat=Parkinson_f;
  • len=size(dataMat,1);
  • %归一化
  • maxV = max(dataMat);
  • minV = min(dataMat);
  • range = maxV-minV;
  • newdataMat = (dataMat-repmat(minV,[len,1]))./(repmat(range,[len,1]));
  • Indices   =  crossvalind('Kfold', length(Parkinson_label), 10);
  • site = find(Indices==1|Indices==2|Indices==3);
  • train_F = newdataMat(site,:);
  • train_L = Parkinson_label(site);
  • site2 = find(Indices~=1&Indices~=2&Indices~=3);
  • test_F = newdataMat(site2,:);
  • test_L =Parkinson_label(site2);
  • save train_F train_F;
  • save train_L train_L;
  • save test_F test_F;
  • save test_L test_L;
  • end6 T5 q4 M1 V4 _9 S! [' Z
   
  x, }( U1 A6 H% Y- ~$ ]$ V
* ]$ j2 A% S2 @8 [' F主函数PSOFS:) ^+ u; i  k) d
, }/ }. R4 H" u: x# n' N5 j
  • clear;
  • clc;
  • format long;
  • %------给定初始化条件----------------------------------------------
  • c1=2;             %学习因子1
  • c2=2;             %学习因子2
  • w=0.7;            %惯性权重
  • MaxDT=100;       %最大迭代次数
  • D=22;             %搜索空间维数(未知数个数)
  • M=30;             %初始化群体个体数目
  • bound=1;
  • %eps=10^(-6);      %设置精度(在已知最小值时候用)
  • global answer      %最后所有粒子的结果(包括特征与精确度)
  • answer=cell(M,3);
  • global choice     %选出的特征个数
  • choice=0.8;
  • %------初始化种群的个体(可以在这里限定位置和速度的范围)------------
  • x=randn(M,D); %随机初始化位置
  • v=randn(M,D); %随机初始化速度
  • x(x>bound)=bound;
  • x(x<-bound)=-bound;
  • %------先计算各个粒子的适应度,并初始化p(i)和gbest--------------------
  • divide_datasets();
  • for i=1:M
  •     p(i)=fitness(x(i,:),i);
  •     y(i,:)=x(i,:);
  • end
  • gbest=x(1,:);             %gbest为全局最优
  • for i=2:M
  •     if(fitness(x(i,:),i)<fitness(gbest,i))
  •         gbest=x(i,:);
  •     end
  • end
  • %------进入主要循环,按照公式依次迭代,直到满足精度要求------------
  • for t=1:MaxDT
  •     for i=1:M
  •         v(i,:)=w*v(i,:)+c1*rand*(y(i,:)-x(i,:))+c2*rand*(gbest-x(i,:));
  •         x(i,:)=x(i,:)+v(i,:);
  •         if fitness(x(i,:),D)<p(i)
  •             p(i)=fitness(x(i,:),i);
  •             y(i,:)=x(i,:);
  •         end
  •         if p(i)<fitness(gbest,i)
  •             gbest=y(i,:);
  •         end
  •     end
  • end
  • %------显示计算结果
  • disp('*************************************************************')
  • Solution=gbest';
  • Result=fitness(gbest,i);
  • disp('*************************************************************')
    6 o6 e$ r9 z3 z& n
           
5 ]2 s& q. R, x
! v) T! d) a  B2 Q7 F2 F+ H特征选择评价函数(利用林志仁的SVM进行训练):
8 a( V& J2 s) Q* R% U
, D. p% p6 K; W' @
  • function error = fitness(x,i)
  • global answer
  • global choice
  • load train_F.mat;
  • load train_L.mat;
  • load test_F.mat;
  • load test_L.mat;
  • inmodel = x>choice;%%%%%设定恰当的阈值选择特征
  • answer(i,1)={sum(inmodel(1,:))};
  • model = libsvmtrain(train_L,train_F(:,inmodel), '-s 0 -t 2 -c 1.2 -g 2.8');
  • [predict_label, ~, ~] = libsvmpredict(test_L,test_F(:,inmodel),model,'-q');
  • error=0;
  • for j=1:length(test_L)
  •     if(predict_label(j,1) ~= test_L(j,1))
  •         error = error+1;
  •     end
  • end
  • error = error/length(test_L);
  • answer(i,2)={error};
  • answer(i,3)={inmodel};
  • end
    9 e6 m1 _8 O+ A. s, _
( m# ~+ M8 D2 v1 b& j
结果(选出的特征数和错误率):! S! v4 _% }8 I) ?9 h
6 M4 M7 L. X4 v$ d7 T$ R* r
4 B$ }! R5 ~6 Y, z' B7 Q) U) P

该用户从未签到

2#
发表于 2020-10-20 20:13 | 只看该作者
PSO算法特征选择MATLAB实现
您需要登录后才可以回帖 登录 | 注册

本版积分规则

关闭

推荐内容上一条 /1 下一条

EDA365公众号

关于我们|手机版|EDA365电子论坛网 ( 粤ICP备18020198号-1 )

GMT+8, 2025-7-23 04:50 , Processed in 0.125000 second(s), 26 queries , Gzip On.

深圳市墨知创新科技有限公司

地址:深圳市南山区科技生态园2栋A座805 电话:19926409050

快速回复 返回顶部 返回列表