聚类算法之k-means的实现

聚类算法是给一大堆原始数据,然后通过算法将其中具有相似特征的数据聚为一类。

k-means聚类,也叫k均值聚类,要先给出原始数据所含的类数,然后将含有相似特征的数据聚为一个类中。并不需要关心某一类是什么,我们需要实现的目标只是把相似的东西聚到一起,不需要使用训练数据进行学习,所以这属于无监督学习。

步骤一般如下:

1、从所以数据中随机取k个元素,作为k个类的各自的中心。

2、分别计算剩下的元素到k个类中心的相异度(常用欧式距离),将这些元素分别划归到相异度最低(即距离最近)的类。

3、根据聚类结果,找到每一类的所有数据,计算他们的平均值,作为下次计算的聚类中心。

4、将数据中全部元素按照新的中心重新聚类。

5、重复第4步,直到聚类结果不再变化。

6、将结果输出。

下面举个实例:

啤酒根据发酵方式分为三大类:

淡色啤酒,浓色啤酒,黑啤酒,其他啤酒对以下数据进行聚类

clear all

clc

data=[144.00 19.00 4.70 ;

181.00 19.00 4.90 ;

157.00 15.00 4.90 ;

170.00 7.00 5.20 ;

152.00 11.00 5.00 ;

145.00 23.00 4.60 ;

175.00 24.00 5.50 ;

149.00 27.00 4.70 ;

99.00 10.00 4.30 ;

113.00 6.00 3.70 ;

140.00 16.00 4.60 ;

102.00 15.00 4.10 ;

135.00 11.00 4.20 ;

150.00 19.00 4.70 ;

149.00 6.00 5.00 ;

68.00 15.00 2.30 ;

136.00 19.00 4.40 ;

144.00 24.00 4.90 ;

72.00 6.00 2.90 ;

97.00 7.00 4.20 ];

% data=[data1;data2;data3;data4];

[row,col] = size(data);

K = 4;

max_iter = 300;%%迭代次数

min_impro = 0.1;%%%%最小步长

display = 1;%%%判定条件

center = zeros(K,col);

U = zeros(K,col);

%% 初始化聚类中心

mi = zeros(col,1);

ma = zeros(col,1);

for i = 1:col

mi(i,1) = min(data(:,i));

ma(i,1) = max(data(:,i));

center(:,i) = ma(i,1) - (ma(i,1) - mi(i,1)) * rand(K,1);

end

%% 开始迭代

for o = 1:max_iter

%% 计算欧氏距离,用norm函数(即向量每个元素分别平方后求和再开方)

for i = 1:K

dist{i} = [];

for j = 1:row

dist{i} = [dist{i};data(j,:) - center(i,:)];

end

end

minDis = zeros(row,K);

for i = 1:row

tem = [];

for j = 1:K

tem = [tem norm(dist{j}(i,:))];

end

[nmin,index] = min(tem);

minDis(i,index) = norm(dist{index}(i,:));

end

%% 更新聚类中心

for i = 1:K

for j = 1:col

U(i,j) = sum(minDis(:,i).*data(:,j)) / sum(minDis(:,i));

end

end

%% 判定

if display

end

if o >1,

if max(abs(U - center)) < min_impro;

break;

else

center = U;

end

end

end

%% 返回所属的类别

class = [];

for i = 1:row

dist = [];

for j = 1:K

dist = [dist norm(data(i,:) - U(j,:))];

end

[nmin,index] = min(dist);

class = [class;data(i,:) index];

end

%% 显示最后结果

[m,n] = size(class);

figure;

title('聚类结果');

hold on;

for i=1:row

if class(i,end)==1

plot3(class(i,1),class(i,2),class(i,3),'ro');

elseif class(i,end)==2

plot3(class(i,1),class(i,2),class(i,3),'go');

elseif class(i,end) == 3

plot3(class(i,1),class(i,2),class(i,3),'bo');

end

end

grid on;

得到结果:

这20种啤酒分别属于类别1,2,1,2,1,1,2,1,3,4,1,3,4,1,1,3,1,1,3,3

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

  • 背景 一年多以前我在知乎上答了有关LeetCode的问题, 分享了一些自己做题目的经验。 张土汪:刷leetcod...
    土汪阅读 12,881评论 0 33
  • 回溯算法 回溯法:也称为试探法,它并不考虑问题规模的大小,而是从问题的最明显的最小规模开始逐步求解出可能的答案,并...
    fredal阅读 13,985评论 0 89
  • Java经典问题算法大全 /*【程序1】 题目:古典问题:有一对兔子,从出生后第3个月起每个月都生一对兔子,小兔子...
    赵宇_阿特奇阅读 2,068评论 0 2
  • 仿佛是一夜间 冷秋覆盖了炎夏 不知哪个方向的风 捎带着寒意 封锁了叶面上浓郁的绿 正午嚣张的闷热 亦或是夜幕下 让...
    夏夜冬日阅读 263评论 2 1
  • 主讲老师:高志鹏 解决问题的关键因素方法、合作、经验。 问题解决模型的四大板块Pass:问题界定,原因分析,方案确...
    商凌云阅读 423评论 1 2

友情链接更多精彩内容