今天了解了一下K均值聚类法,编了一个matlab的实现。
首先,随机产生一组随机点,然后任意选择其中的两个,将其作为初始类心,然后使用K均值的方法循环完成聚类。当然同样的方法也适用于多类的情况。
这个就当是学习后的一个作业吧。
第一步:生成随机点
clear all;clc;
c=randn(1,50);
d=randn(1,50);
x=[c;d];
a=randn(1,50)+3;
b=randn(1,50)+3;
y=[a;b];
y=[x y];
for ii=1:2*length(a)
plot(y(1,ii),y(2,ii),'r.','Markersize',20);hold on;
end
plot(y(1,1),x(2,1),'b.','Markersize',20);hold on;
plot(y(1,8),x(2,8),'g.','Markersize',20);hold on;
第二步:K均值聚类
zs1=y(:,1);
zs2=y(:,8);
z1=[0;0];
z2=[0;0];
zb1=[0;0];
zb2=[0;0];
cont1=0;
cont2=0;
while(zs1~=zb1)
for ii=1:2*length(a)
jj1=norm(zs1-y(:,ii));
jj2=norm(zs2-y(:,ii));
if jj1 < jj2
plot(y(1,ii),y(2,ii),'b.','Markersize',20);hold on;
z1=z1+y(:,ii);
cont1=cont1+1
else plot(y(1,ii),y(2,ii),'g.','Markersize',20);hold on;
z2=z2+y(:,ii);
cont2=cont2+1
end
end
zb1=zs1;
zb2=zs2;
zs1=z1/cont1;
zs2=z2/cont2;
cont1=0;
cont2=0;
z1=[0;0];
z2=[0;0];
end