数学建模社区-数学中国
标题:
KNN分类算法及MATLAB程序与结果
[打印本页]
作者:
杨利霞
时间:
2021-4-9 11:18
标题:
KNN分类算法及MATLAB程序与结果
8 @) e- {4 i( j/ [5 n. Z k
KNN分类算法及MATLAB程序与结果
K邻近算法
# w& Y: b( k. h- e
' L9 V% B' C) W5 a) K, v% b: q
KNN算法的决策过程
! Y- \. k7 y c
k-Nearest Neighbor algorithm
4 H1 B w0 c0 {! o6 Q6 c
右图中,绿色圆要被决定赋予哪个类,是红色三角形还是蓝色四方形?如果K=3,由于红色三角形所占比例为2/3,绿色圆将被赋予红色三角形那个类,如果K=5,由于蓝色四方形比例为3/5,因此绿色圆被赋予蓝色四方形类。
. ~: F: l6 n1 X( B8 K+ w
K最近邻(k-Nearest Neighbor,KNN)分类算法,是一个理论上比较成熟的方法,也是最简单的机器学习算法之一。该方法的思路是:如果一个样本在特征空间中的k个最相 似(即特征空间中最邻近)的样本中的大多数属于某一个类别,则该样本也属于这个类别。KNN算法中,所选择的邻居都是已经正确分类的对象。该方法在定类决 策上只依据最邻近的一个或者几个样本的类别来决定待分样本所属的类别。 KNN方法虽然从原理上也依赖于极限定理,但在类别决策时,只与极少量的相邻样本有关。由于KNN方法主要靠周围有限的邻近的样本,而不是靠判别类域的方 法来确定所属类别的,因此对于类域的交叉或重叠较多的待分样本集来说,KNN方法较其他方法更为适合。
4 Z! j1 ?9 w1 o, n0 P2 A4 N7 l
KNN算法不仅可以用于分类,还可以用于回归。通过找出一个样本的k个最近邻居,将这些邻居的属性的平均值赋给该样本,就可以得到该样本的属性。更有用的方法是将不同距离的邻居对该样本产生的影响给予不同的权值(weight),如权值与距离成正比。
9 s" ~7 A! e5 }2 y" F/ c9 O3 s0 g7 V. N
该算法在分类时有个主要的不足是,当样本不平衡时,如一个类的样本容量很大,而其他类样本容量 很小时,有可能导致当输入一个新样本时,该样本的K个邻居中大容量类的样本占多数。因此可以采用权值的方法(和该样本距离小的邻居权值大)来改进。该方法 的另一个不足之处是计算量较大,因为对每一个待分类的文本都要计算它到全体已知样本的距离,才能求得它的K个最近邻点。目前常用的解决方法是事先对已知样 本点进行剪辑,事先去除对分类作用不大的样本。该算法比较适用于样本容量比较大的类域的自动分类,而那些样本容量较小的类域采用这种算法比较容易产生误 分。
( A$ k2 F& b/ }% W8 I3 ~5 q
2 d. H+ _$ Y- x2 w
相比于之前介绍的SVM及BP算法,KNN算法思想比较简单,它没有训练出模板model,然后利用model进行测试的过程,而是直接就是训练,利用已知的有标记的样本,进来一个样本就计算它与这些已知样本的距离,然后对这些距离中挑K个最小的进行投票,看K个最接近的样本中哪一个样本所占的个数多,就判断为哪一类。
5 |2 }2 |: W! g D+ A
6 i! W- I0 d) {: o' l
前面也提到了KNN算法的一个当样本不平衡时,如一个类的样本容量很大,而其他类样本容量 很小时,有可能导致当输入一个新样本时,该样本的K个邻居中大容量类的样本占多数,为此我觉得可以在每类中随机抽取相同的样本数进行重组成新的样本,然后再进行KNN算法,记为改进方法1,或者调节距离的权重,距离越小的投票时的权重值就越大,记为改进方法2。下面是KNN算法的MATLAB代码,及没有用随机抽样的方法进行的试验结果(三类数据分别为150 50 50),和改进方法1,及改进方法2的结果对比。
! t3 g( G/ S$ ]1 Y# H' E' P, ?
$ ^: O q* ]& [2 I6 d; e0 q
未改进的结果,由于其中一类样本是150个,其他两类是50,所以数据有偏斜,结果为:
6 c; d# N- u; F9 g% o
( b, y2 y2 a) t
整体分类精度为:0.90476
. y: x! j6 v% w H/ E+ b# {
类别1的分类精度为:0.94118
* W; z$ f& C) w3 k
类别2的分类精度为:0.58824
( t# }# r& s2 k" k; v
类别3的分类精度为:1
, ^ |; b- ` `5 Z6 E' L* |
$ L% j5 x( w2 P8 L
# n2 {# {: d& Z0 b8 Q# \) k
改进方法一,随机抽样组成新的样本,新样本中,各类的样本数相同。结果为:
2 t2 }8 u5 H6 B0 z* l
5 z: N4 P$ ?7 h4 s9 L8 J) c
整体分类精度为:0.94048
. t# u- d7 u' e6 b
类别1的分类精度为:0.94118
5 j# {8 h/ A6 f) i9 T
类别2的分类精度为:0.76471
! m0 [8 H! N. ~8 K3 [* ^& o1 \" ~
类别3的分类精度为:1
+ d2 r& r# {0 l" W1 u
改进方法二,增大最小距离的投票权重,当然也可能是我的权重选的不合适,结果为:
# n# f& t( B, T( A2 G( {
- d* x# P3 h* J6 e( W1 \
整体分类精度为:0.90476
J. x. W f/ \ r
类别1的分类精度为:0.94118
$ u4 q% d" c7 ]* H' u
类别2的分类精度为:0.58824
( R) p7 z1 W$ ^2 G- J
类别3的分类精度为:1
- Q% z- K, i7 h* X
可见和没加权重的未改进结果相同,当然我也试过好多数值,如果不当,精度还会下降。
1 x; X3 E1 ^3 M7 `! Y" j8 v$ s
% X6 l4 p$ I2 J7 B% s j7 I
把改进方法一和改进方法二相结合的结果:
4 Y% l2 G5 ]1 {
f, t" Y3 B8 q4 U
整体分类精度为:0.90476
" K5 c# Y% O% t* t' g
类别1的分类精度为:0.94118
. R! W& f8 V6 O4 n$ L' z8 W
类别2的分类精度为:0.76471
8 N+ P3 @1 g+ n$ j$ J
类别3的分类精度为:0.94
+ `- N" O6 E0 v% c' S3 {
只是类别三的精度相对于改进方法一有所下降,所以看来还是样本数据的分均匀程度对分类的结果影响比较大。
4 j* n$ s5 J. u# N7 B3 l, g8 s' A
" b- T6 i* W- z1 S" W; J7 y7 k
下面是程序的源代码:
L9 F, y) c& N, q/ q% b! r
2 c1 ?2 Y/ t& W l) X4 j z
clear all;
5 W8 l1 H- y$ S' ~
close all;
% m& _" J/ n* E1 T( d- E5 g2 D/ c) y
clc;
$ l- z* h4 W( o7 L2 ?0 f
load D:\硕士学习资料3\硕士毕设资料\区域生长等分割算法程序\遗传分割\MYGAseg\特征选择\traindata.mat
% {! {* k$ p0 K* G( w
load D:\硕士学习资料3\硕士毕设资料\区域生长等分割算法程序\遗传分割\MYGAseg\特征选择\testdata.mat
' I' p& l9 A9 g$ T5 k/ q$ d( H8 u
load D:\硕士学习资料3\硕士毕设资料\区域生长等分割算法程序\遗传分割\MYGAseg\特征选择\trainlabel.mat
1 U1 ?/ M$ q6 E) D
load D:\硕士学习资料3\硕士毕设资料\区域生长等分割算法程序\遗传分割\MYGAseg\特征选择\testlabel.mat
0 m. W% s/ G5 b
traindata=traindata(1:99,
;
1 T# C/ j" y7 \# `
trainlabel=trainlabel(1:99,1);
+ l: c' W* L1 q; z- O" ^" c
[predictlabel,acc]=myKNN(traindata,trainlabel,testdata,testlabel,15);
8 T' Z3 U$ {3 P0 Q3 ?- k
% \1 p& U! |" \2 J
* W7 k5 C& I' O$ Y4 A, q4 Z
function [predictlabel,acc]=myKNN(traindata,trainlabel,testdata,testlabel,K)
- d- ?' W4 y) |4 Q* k
[mtrain,ntrain]=size(traindata);
$ z$ w {5 i9 j# h7 n; B
[mtest,ntest]=size(testdata);
4 `& p) u: t6 f7 e0 M, W
juli=zeros(mtest,mtrain);%每一行存储的是一个样本与所有训练样本的距离
% {4 n0 h" z* V, k& I
sortjuli=zeros(mtest,mtrain);
: r! M% d# R' z
zuobiaojuli=zeros(1,mtest);
6 c0 A1 c+ u' K* r% \7 V- I2 u
num1=0;
) |& D* C; @1 l1 Q: O/ F. B
num2=0;
7 t+ j! @' |6 U" L# J& g
num3=0;
+ `! P9 t# x% c& j) b" D1 B
rightnum=0;%记录整体分类正确的个数
& q6 `2 a6 |5 h1 c# j) W
rightnum1=0;
# A, p3 q6 @1 v" D
rightnum2=0;
0 i9 ]8 ]/ d* ~
rightnum3=0;
; Y, j) c, @6 f6 X" ]+ n" C
for i=1:mtest
' `# f* {- I& u% Q" I) t
for j=1:mtrain
% p7 i' T% m. h/ r
juli(i,j)=sqrt((sum(testdata(i,
-traindata(j,
).^2));%计算每个测试样本与每个训练样本的oushi距离
# k2 S9 n$ g( }6 l7 p5 ~1 b
end
2 n& e3 t) N6 h1 E$ H
end
% q5 R3 L( G A: ~' p
for i=1:mtest
4 g j) k: m. j
[ sortjuli,zuobiaojuli]=sort(juli(i,
);%sort函数默认返回的是由小到大的排序结果
/ `7 q( c T* u/ m! V7 B) A, M
sorttrainlabel=trainlabel(zuobiaojuli);%标签也相应的改变顺序
. y3 m6 [6 V5 }/ L4 c8 _" r
ksortjuli(1,
=sortjuli(1,1:K);%取距离和坐标的前k个
* p; d! ~) X- p% U- P) P
ksorttrainlabel=sorttrainlabel(1:K,1);
3 r: c; r0 B- {0 r. @ a
for j=1:K%统计前K个结果中那个的数目多
; x- I5 p @' Y- x# V
if j==1%如果是排在第一位的,也就是距离最小的,那么投票权重加大4
9 u: n' y u+ C: o) T% @
if ksorttrainlabel(j,1)==1
8 W) V0 F* p: g
num1=num1+5;
- q& Q) z# d w8 O( h% c. R- X
end
+ U3 o- M4 C1 p
if ksorttrainlabel(j,1)==2
9 ~4 s0 s; p0 H2 T1 t
num2=num2+5;
" w. @& j3 H$ N
end
4 ~; V# N' ], T/ E2 Z. n2 t
if ksorttrainlabel(j,1)==3
, q0 Y+ _0 A1 E2 \
num3=num3+5;
! f8 D. R \# w F
end
- F$ E$ C# T8 B+ a
elseif j==2%排在第二位的权重加大2,当然也可以自行改变
; d7 Y7 ` y1 L! V
if ksorttrainlabel(j,1)==1
( m U- O V- Y3 u
num1=num1+2;
( c! b R4 r3 B
end
4 v6 l' z" g+ p' \) `& U: v
if ksorttrainlabel(j,1)==2
( E4 L& X1 z# `# S: s1 r
num2=num2+2;
- O; ]+ D% X( F
end
+ H* d6 Y/ C6 Y9 h7 P q
if ksorttrainlabel(j,1)==3
% } {2 ^0 @, E- o0 N
num3=num3+2;
+ r8 F3 W$ X) ^# O; @1 J6 ]& V
end
' _! B* O5 C9 u* ^- z2 F
else
5 R6 r+ l3 W6 z! c/ A
if ksorttrainlabel(j,1)==1
w8 c* x3 y" J% W; z6 W5 {/ p
num1=num1+1;
) c5 }' j* ^& E! |+ P
end
6 t D7 o# b: R: V; ^
if ksorttrainlabel(j,1)==2
/ ~7 f( {8 v+ B/ V& a: N2 [1 Z
num2=num2+1;
. r9 F: i+ c6 v9 X9 k& T
end
( A( M1 e5 B [4 y: o1 \
if ksorttrainlabel(j,1)==3
8 U6 C$ H" R% R( e+ R* }, B
num3=num3+1;
: u8 t$ a8 C1 w; B9 K6 J, M. v
end
7 G- a, ]/ J' K
end
" X- L" D; r5 U9 z4 K
end
$ G8 T5 Z; _# I; `
A=[num1 num2 num3];
2 s* X- S% d4 f. R% V
Xsort=sort(A);%按照从小到大排序
) ^7 }: X. W: ^. u% e
if Xsort(1,end)==num1
! d! m/ G! j- M+ ~9 W
predictlabel(i,1)=1;
3 j9 ^. B- G8 @/ J) a2 H
elseif Xsort(1,end)==num2
) ]0 [. g! n& |( T* \* j0 F& L9 v9 U
predictlabel(i,1)=2;
$ X% T3 k( @ A1 p8 F; U7 p+ }5 ~
elseif Xsort(1,end)==num3
5 @8 z& ~5 B9 K4 s! ]
predictlabel(i,1)=3;
( ^$ X& Q. F- ` C7 _% T
end
4 g2 E6 k' ^4 K$ _3 c# k7 ]
if predictlabel(i,1)==testlabel(i,1)
5 W# u3 `7 z, }4 ?- E4 z# b, F. {$ p
rightnum=rightnum+1;
, ]- ~' G J( B& p3 _ Y* t( R
end
, W/ A0 Y3 }4 S2 |) k
if predictlabel(i,1)==testlabel(i,1)&&testlabel(i,1)==1
! X( K5 o$ C4 Q5 u: w& G1 s/ K
rightnum1=rightnum1+1;
) S' y& U4 ]/ T# Y
end
9 |& {' W) {: S' M+ ` y0 ]; L
if predictlabel(i,1)==testlabel(i,1)&&testlabel(i,1)==2
`. C0 w9 P: Y7 ^
rightnum2=rightnum2+1;
H0 Q- u. }: D2 t1 x; b! {
end
f9 ~. L7 B d+ k* l. ]! u
if predictlabel(i,1)==testlabel(i,1)&&testlabel(i,1)==3
1 O. X2 X6 ^7 R
rightnum3=rightnum3+1;
' M3 |9 S- [4 R+ d4 p, i* Z
end
1 G: }1 M3 `: \7 [) u
/ E' o4 W t4 Q
num1=0;%清0
3 }, Z" X( {5 x. i
num2=0;
z0 F) P) V; L" G. y
num3=0;
) R# ]+ @0 U. t# |+ s
end
% q4 q! H' }7 ^& v5 {0 g
leibie1=size(find(testlabel==1),1);%找出测试样本1的个数
( h: I( i/ B# ]
leibie2=size(find(testlabel==2),1);
$ x9 b# F: J! K7 m; K) D
leibie3=size(find(testlabel==3),1);
3 \' M# [% E* j; P
acc1=rightnum1/leibie1;
5 F0 \+ q( S- B `+ e
acc2=rightnum2/leibie2;
8 w! N" S: [# O
acc3=rightnum3/leibie3;
# r; p) z8 G& j
acc=rightnum/mtest;%总体分类精度
8 h. S, r9 y* r0 X: a
disp(['整体分类精度为:',num2str(acc)]);
0 b+ X' l6 H$ n) Z5 I$ E0 P
disp(['类别1的分类精度为:',num2str(acc1)]);
' Q4 I8 o# R! y9 y5 A/ z
disp(['类别2的分类精度为:',num2str(acc2)]);
+ u% _. Q- j) L! |
disp(['类别3的分类精度为:',num2str(acc3)]);
; q2 I3 Y ^4 B' e1 }
9 a& i" E/ m3 L2 s1 y
- C$ }) Q( e, E+ A& l+ C) h" |3 W
K最邻近密度估计技术是一种分类方法,不是聚类方法。
* T" ^+ k0 A6 _: q4 C0 k4 d0 ]6 _# e
% `/ S: Y( @8 M, e% d4 N
不是最优方法,实践中比较流行。
8 y7 Q2 ?0 Q! y1 W4 {
3 l$ U& G/ `$ X* z8 A) x4 {" b
通俗但不一定易懂的规则是:
/ _) S$ c6 w2 |" E% H
6 ]+ _ Q b' d! e3 ^6 e. S$ [/ m
1.计算待分类数据和不同类中每一个数据的距离(欧氏或马氏)。
' T1 \5 p0 i% c2 P( x( ~
* V2 X/ N& S: b( t( `3 b* w c: r
2.选出最小的前K数据个距离,这里用到选择排序法。
# I8 Z3 ?% c* e! C
0 W! A& q$ q5 V, } P7 j6 H
3.对比这前K个距离,找出K个数据中包含最多的是那个类的数据,即为待分类数据所在的类。
) P6 g1 i" ]# i" }. x
) ?1 X% S* l1 q$ Q
不通俗但严谨的规则是:
8 A" }# |0 F$ A0 I1 X# }7 B1 x0 }
& \$ v" C0 @) x+ @4 [
给定一个位置特征向量x和一种距离测量方法,于是有:
( Z0 S- E' Q) e) K; y
( k0 I/ l* x$ x, f/ h
1.在N个训练向量外,不考虑类的标签来确定k邻近。在两类的情况下,k选为奇数,一般不是类M的倍数。
9 Z! V; R, ]* i F
% G2 T7 a9 y( O: n
2.在K个样本之外,确定属于wi,i=1,2,...M类的向量的个数ki,显然sum(ki)=k。
& r& k( X7 _" v$ l! L& o
' G; ]! g) t9 \9 ]. `
3.x属于样本最大值ki的那一类wi。
4 W# [/ y# v6 N5 \8 T: g! k) ^$ G- A
————————————————
2 n* z( f5 ?3 \# p5 S% l
版权声明:本文为CSDN博主「dulingtingzi」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
s/ m' x" h3 X: ~% {/ }
原文链接:https://blog.csdn.net/dulingtingzi/article/details/51152363
* |- _- A- G7 t' M( f) q# E
e6 }8 a- f+ F
: J6 L7 X+ f5 v& Y& J2 G# ]: D# Q- G
欢迎光临 数学建模社区-数学中国 (http://www.madio.net/)
Powered by Discuz! X2.5