数学建模社区-数学中国
标题:
【聚类算法】带你轻松搞懂K-means聚类(含代码以及详细解释)
[打印本页]
作者:
杨利霞
时间:
2022-9-13 12:26
标题:
【聚类算法】带你轻松搞懂K-means聚类(含代码以及详细解释)
【聚类算法】带你轻松搞懂K-means聚类(含代码以及详细解释)
' X/ |! p* C- k* U0 W k- k
9 [5 L! W3 V0 r& E
文章目录
$ d* K+ S' c6 w& T
一:K-means聚类算法
+ Z/ f; i( U' ^8 {$ S. ?
二:实例分析
7 O2 v. J( i9 g! K, Q( M
三:原理与步骤
* t' l' k. I# L4 `$ D1 q3 r
四:Matlab代码以及详解
1 A! g( @4 C' P; N5 N: r
一:K-means聚类算法
7 ^. i. M0 S4 ~ ~% V: A
聚类是一个将数据集中在某些方面相似的数据成员进行分类组织的过程,聚类就是一种发现这种内在结构的技术,聚类技术经常被称为无监督学习。
3 b, N* u3 v [, ?9 R0 `4 D7 N) q5 A) ?
k均值聚类是最著名的划分聚类算法,由于简洁和效率使得他成为所有聚类算法中最广泛使用的。给定一个数据点集合和需要的聚类数目k,k由用户指定,k均值算法根据某个距离函数反复把数据分入k个聚类中。
% f" _7 r M3 g7 C4 X4 I& j
! ]) w0 ~, {' _
二:实例分析
! L/ ?% C8 g) Y! P, \+ l
现有50个二维数据点如下图,使用K-Means算法将以下数据实现聚类。
! N0 g$ ^* A! V# a5 v9 t2 S
, C' H7 p3 t# n) W1 S+ H
结果展示:
! F$ F/ B4 R8 o" G' {! P- ?* p# c; q
* v& o8 C# q- e* |' T: y# [( L
4 t6 H! V0 L. ?& y1 _
三:原理与步骤
9 m& l0 r9 y, S( v& n# ]
K-means算法是典型的基于距离(欧式距离、曼哈顿距离)的聚类算法,采用距离作为相似性的评价指标,即认为两个对象的距离越近,其相似度就越大。该算法认为簇是由距离靠近的对象组成的,因此把得到紧凑且独立的簇作为最终目标。
8 t7 {$ \* J5 b! `3 l0 Q
K-mean算法步骤如下:
0 y9 G# y( i2 K) M" Q, l
: Z4 L6 O( W- t" I: b1 h- K
先定义总共有多少个簇类,随机选取K个样本为簇中⼼。
5 D3 w3 _& K# ~4 {& e$ H5 v4 V: k: F
分别计算所有样本到随机选取的K个簇中⼼的距离。
# Z" c3 z* z+ z/ |
样本离哪个中⼼近就被分到哪个簇中⼼。
8 C2 M) \- q6 W. j
计算各个中⼼样本的均值(最简单的⽅法就是求样本每个点的平均值)作为新的簇心。
3 G- M7 U; N, C" B7 _4 t' ^5 w
重复2、3、4直到新的中⼼和原来的中⼼基本不变化的时候,算法结束。
& Z: W* C3 f+ b- b: H
算法结束条件:
# C$ w. `6 W% _8 d
6 f7 O# W# M. ~0 |7 W+ K
当每个簇的质心,不再改变时就可以停止k-menas。
+ B7 o, n1 C2 ~, j
当循环次数达到事先规定的次数时,停止k-means
2 ]+ Y- j4 L8 G) F$ Q
原理示意图:
9 L& P r( [0 h% T2 x; S! @4 |
5 J1 {1 I' e, w2 e# N
简单小实例:
$ A. `; S: U% H2 g& a6 e& K( y
有以下6个点,初始随机选取两个点作为两个簇的簇中心(这里假设选取的是A3,A4),求最后的簇所属情况。
; C" d( O1 `; M9 z; w
$ x k9 n: ~+ x; h( f6 U; z, F
1️⃣:计算每个点到簇心的距离,将距离近的归为一类。
0 L# x S* w* s9 [4 O( P
7 V- d) ~3 b; z) j- ~9 Q- L
2️⃣:将红色对应的点和绿色对应的每个点分别求X,Y平均值,最为新的簇心。
; w \; \1 ]. @6 h
: c4 Z0 U: d. R$ y- Q5 [' n
3️⃣:计算每个点到新簇心的距离,继续将对应距离近的点归为一类。
5 Q: ?4 f; M$ R! j
, E" x2 h* t% S1 F4 ]4 ~/ G. A
4️⃣:由于关联点没有发生变化,所以之后的结果不会发生变化。停止计算
2 H$ S' U* U9 R( q2 X( x+ J' J! H
5️⃣:得结果红色簇:A1,A3,A5,紫色簇:A2,A4,A6。
" j: x8 c8 K; o9 [& b" z# M6 I" k. w
5 a2 S, u j! h1 g# C
四:Matlab代码以及详解
: V4 v1 M4 M7 Z. `3 |- Q5 G3 y
clc;clear;close all;
$ g; b; B! I2 T6 W
data(:,1)=[90,35,52,83,64,24,49,92,99,45,19,38,1,71,56,97,63,...
/ C& p O; Q6 f+ D: W" E
32,3,34,33,55,75,84,53,15,88,66,41,51,39,78,67,65,25,40,77,...
0 G0 X. U5 z( ~% H
13,69,29,14,54,87,47,44,58,8,68,81,31];
9 H( i2 S: G5 A
data(:,2)=[33,71,62,34,49,48,46,69,56,59,28,14,55,41,39,...
! a+ q( D3 |0 C8 J) m+ G5 U
78,23,99,68,30,87,85,43,88,2,47,50,77,22,76,94,11,80,...
+ d7 K) c& m+ p4 Z
51,6,7,72,36,90,96,44,61,70,60,75,74,63,40,81,4];
* ^: ^* i$ I' ^' Y% h m* D$ w% k
%50 * 1
6 ^8 Z6 H( U: E6 P/ n$ Q
figure(1)
: v$ u# B" a, b J( ^
; q' \3 t7 t L; k* O
scatter(data(:,1),data(:,2),'MarkerEdgeColor','r','LineWidth',2)
, y4 G2 l' V$ ^! d" l
%% 原理推导K均值
/ ?( H) b2 }+ h( B5 o. T0 }5 y
[m,n]=size(data);%m = 50,n = 1;
5 b1 n' E0 O: w: V/ o
cluster_num=4;%4个初始中心
; O! E- k2 t: O. m$ b6 [
cluster=data(randperm(m,cluster_num),
;%randperm(m,cluster_num)在前m中随机选取cluster_num个 %随机选取中心
6 {, x3 D2 n7 U, F5 T! q7 T8 O
%data函数 取数据用
2 B7 f8 @* U" S3 {( W. f! s
epoch_max=1000;%最大次数
! H) d4 b6 X- E5 G; q; g
therad_lim=0.001;%中心变化阈值
6 o1 N# N2 R3 E0 b
epoch_num=0;
1 p+ Q S) E% p
while(epoch_num<epoch_max)
, n$ Z' N9 @5 G! J1 V2 m9 [5 @
epoch_num=epoch_num+1;
+ ?( j# B2 j; `
for i=1:cluster_num
9 n, U+ s" \4 m1 R9 n8 B6 N6 T
distance=(data-repmat(cluster(i,
,m,1)).^2;% 50 * 2 repmat扩展矩阵
3 o0 ]: o) K3 X$ y0 W
%.^2是矩阵中的每个元素都求平方,^2是求矩阵的平方或两个相同的矩阵相乘,因此要求矩阵为方阵
7 c9 @8 V( i: j) F
distance1(:,i)=sqrt(sum((distance),2));%求行和
& U7 y2 X2 z. i' q* R. D9 k8 C$ c
%distance1(:,i)=sqrt(sum(distance'));% 默认求列和 1表示每一列进行求和,2表示每一行进行求和;
' k2 w8 l% \ z5 t8 s! ?0 {
%sqrt(sum(distance')) == 1 * 50
/ A( `' [+ C& u7 S' C& t8 @
%distance1 50 * 4 表示每个点距离第i个点的距离
( j5 m3 [# c. o6 x5 I
end
1 j0 J/ |+ ?4 G4 g* m& @
[~,index_cluster]=min(distance1');%distance1' = 4 * 50,min 求列最值 index_cluster = 最小值所在行号 index_cluster = 1 * 50
; I l# Q: J9 {2 ]
for j=1:cluster_num
9 n3 d# Q+ k2 g& F' v
cluster_new(j,
=mean(data(find(index_cluster==j),
);% 4 * 2 找到距离对应中心最近的点 横纵坐标各取平均值
* l, U7 ]1 ?- j' N+ _. e" y; L' U* j
end
' l+ K% p" q+ D% x: `( l
if (sqrt(sum((cluster_new-cluster).^2))>therad_lim)
- x2 B* E. |' D
cluster=cluster_new;
$ v- n" V2 Q' L8 c7 }0 w0 R3 Y
else
. G; S# j+ H9 O" U c1 f7 W) }
break;
/ O4 } z3 v$ d9 g# v
end
+ J6 H1 j/ J) N( E) ~0 ^
end
" D" a3 Y+ g0 [+ l
%% 画出聚类效果
$ Z( d' z( c: b
figure(2)
) Y& y4 {1 _+ ~4 ^
%subplot(2,1,1)
; K4 W$ f; |! q( q
a=unique(index_cluster); %找出分类出的个数
8 i7 K' f% d+ l/ v1 ?' T: b
C=cell(1,length(a));%1 * 4的元胞
0 a2 H: s/ w$ j& k5 u
for i=1:length(a)
. _* ]; ^6 L$ ^) T6 R
C(1,i)={find(index_cluster==a(i))};
; a" a7 w2 `/ L3 _4 C" W
end
7 i4 q* E/ C0 X! w% x' v% t1 C3 H
for j=1:cluster_num
, t; k7 [% m/ K7 e
data_get=data(C{1,j},
;%从data中取每个类的点
) a8 ]3 I! Y, V3 K$ \( K
scatter(data_get(:,1),data_get(:,2),80,'filled','MarkerFaceAlpha',.6,'MarkerEdgeAlpha',.9);
; g4 N- M) s/ F
hold on
0 p+ o8 {' z6 g+ p
end
1 g N3 A0 C7 o
plot(cluster(:,1),cluster(:,2),'kp','LineWidth',2);%画出4个聚类中心
& w. A. `& J* V
hold on
; \7 T" _! R& x0 O3 T
sc_t=mean(silhouette(data,index_cluster'));
, A) Q6 B8 }( B% s9 v$ F/ v
title_str=['原理推导K均值聚类',' 聚类数为:',num2str(cluster_num),' SC轮廓系数:',num2str(sc_t)];
6 D3 c# k; | p1 I. @' o% t
title(title_str)
8 \& y: @% E/ ]! y: Z+ { g
' g& F. z. Q1 L/ u
————————————————
; d% ]- ]$ C* {1 m& u! g! y
版权声明:本文为CSDN博主「Rookiep」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
3 h# J6 M: o* n& h6 k
原文链接:https://blog.csdn.net/qq_43727529/article/details/126813321
6 F* X" D+ z1 p3 M6 ~6 y$ M# @
5 x9 k: [: t& @# d& L. ]% }
0 U8 Y; u- g. c5 @* I% c
欢迎光临 数学建模社区-数学中国 (http://www.madio.net/)
Powered by Discuz! X2.5