fisher算法 matlab实现（分类线、投影方向、取点）

%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%分别产生x轴和y轴都为正态分布的随机序列
%假设x轴和y轴序列相互独立，可产生二维正态分布随机序列
%w1、w2分别用来保存两个训练集的横坐标和纵坐标
%用normrnd函数产生正态分布函数
%normrnd（mean,omega,[row,column]）
%mean:均值;omega:标准差
%row:产生随机序列的行数;column:产生随机序列的列数
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
 X1 = normrnd(40,10,[200,1]);
 Y1 = normrnd(40,10,[200,1]);
 w1=[X1, Y1];
 X2 = normrnd(5 ,10,[100,1]);
 Y2 = normrnd(0 ,10,[100,1]);
 w2=[X2, Y2];
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%以下部分为fisher算法的实现
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%计算样本均值
m1=mean(w1)';
m2=mean(w2)';
%s1、s2分别代表表示第一类、第二类样本的类内离散度矩阵
s1=zeros(2);
[row1,colum1]=size(w1);
for i=1:row1
     s1 = s1 + (w1(i,:)'-m1)*(w1(i,:)'-m1)';
end;
s2=zeros(2);
[row2,colum2]=size(w2);
for i=1:row2
     s2 = s2 + (w2(i,:)' - m2)*(w2(i,:)' - m2)';
end;
%计算总类内离散度矩阵Sw
Sw=s1+s2;
%计算fisher准则函数取极大值时的解w
w=inv(Sw)*(m1-m2);
%计算阈值w0
ave_m1 = w'*m1;
ave_m2 = w'*m2;
w0 = (ave_m1+ave_m2)/2;
%画出两类训练样本点
figure(1)
plot(X1,Y1,'.r',X2,Y2,'.b');%画出两类样本点
hold on;grid;
%画出取极大值时的解w,w(2)/w(1)表示斜率，这条直线并不是分类线，画的只是投影方向
x = [-40:0.1:40];
y = x*w(2)/w(1);
plot(x,y,'g--','LineWidth',3)
k=w(2)/w(1);
wk=[1 k];
wk=wk/sqrt(sum(wk.^2));  
A=[w1;w2];
wd= A*wk'*wk;
plot(wd(:,1),wd(:,2),'xb');
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%画红色点划线表示的分类线
x=linspace(-20,60,100);
y=(w0-x*w(1,:))/w(2,:);
plot(x,y,'-.r');
hold on
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
%以下为测试部分
%利用ginput随机选取屏幕上的点（可连续取10个点）
%程序可根据点的位置自动地显示出属于那个类
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
for i=1:10
    [x,y]=ginput(1);
    plot(x,y,'m*');
    sample=[x,y];
    hold all
    if(sample*w - w0>0)
         disp('it belong to the first class');
    else
         disp('it belong to the second class');
    end;
end
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%