按课件写的matlab版k-means聚类程序
num =40;%一类随机点的数目
x1=1+69*rand(1,num); %生成第一类随机点
y1=1+69*rand(1,num);
scatter(x1,y1)
x2=40+60*rand(1,num);%生成第二类随机点
y2=40+60*rand(1,num);
x=[x1,x2];
y=[y1,y2];
x = floor(x);
y = floor(y);
hold on;
scatter(x2,y2)
title('raw data');
r=randi(2*num,1,2);
while r(1)==r(2)
r=randi(2*num,1,2);
end
h1=x(r(1));
h2=y(r(1));
h3=x(r(2));
h4=y(r(2));
h=[h1,h2,h3,h4];%生成随机特征向量作为聚类中心
hn=[0,0,0,0];
while hn ~= h%聚类中心不再发生变化,则停止计算
h = hn;
k1=0;
k2=0;
sum1 = 0;
sum2 = 0;
sum3 = 0;
sum4 = 0;
for i=1:2*num
if ((x(i)-h1)^2 + (y(i)-h2)^2) < ((x(i)-h3)^2 + (y(i)-h4)^2)%不能改成大于号,按最小距离原则分类
sum1 = sum1+x(i);
sum2 = sum2+y(i);
flag(i) = 1;%模式分类标签
k1 = k1+1;
else
sum3 = sum3+x(i);
sum4 = sum4+y(i);
flag(i) = 0;
k2 = k2+1;
end
end
h1 = sum1/k1;
h2 = sum2/k1;
h3 = sum3/k2;
h4 = sum4/k2;
hn=[h1,h2,h3,h4];%迭代一次后的聚类中心
% flag
% scatter(h1,h2,'r*');
% scatter(h3,h4,'g*');
end
hold off;
figure;
hold on
title('result');
for i=1:2*num
if flag(i) == 1
scatter(x(i),y(i),'r*');
else
scatter(x(i),y(i),'k*');
end
end