数学建模社区-数学中国
标题:
K-近邻算法分类和回归
[打印本页]
作者:
杨利霞
时间:
2022-9-5 15:43
标题:
K-近邻算法分类和回归
- C. P6 D2 s0 p
K-近邻算法分类和回归
( ^0 F$ S( o* k; h3 e [/ T7 M3 ]0 U# o
K近邻算法的主要思想是用离测试集数据点最近的训练集点(称为其邻居)的输出来估计测试集数据点的输出，参数K代表用多少个邻居来估计。超参数K通常设置为奇数来防止平局现象。
2 |, N$ ]* N e2 F& Q5 |. G
! h3 P A" g5 C) b) |, J8 b7 g
其中对邻居的判定：我们可以用欧几里得距离来衡量距离来确定其K个邻居。
" q3 T0 }! N/ L
9 D; n7 v. l2 y
K近邻算法是一种惰性学习和非参数模型。当训练数据数量庞大，同时你对响应变量和解释变量之间的关系所知甚少时，非参数模型会非常有用。KNN 模型只基于一个假设：互相接近的实例拥有类似的响应变量值。非参数模型提供的灵活性并不总是可取的，当训练数据很缺乏或者你对响应变量和解释变量之间的关系有所了解时，对响应变量和解释变量之间关系做假设的模型就很有用。
9 ]- q! O: g5 a }6 E
3 y6 p- v' ^* h# d; N7 \, r
KNN模型分类：
) C. z* |0 q4 g8 K- I1 v3 u0 _
下面我们看一个分类的例子和代码实现来了解一下K近邻算法：
F H; q. j- ~
0 a7 X3 K, N" U D+ b/ X# z/ Z
/ s* c: v/ u) f1 m
. C& a9 p7 Z& t, _1 [
上表是我们的训练集，下面先对数据进行可视化
b. j) [ f1 O @* r- Y
3 E3 c: c& P- o1 o3 [, U; r
import numpy as np
" ?/ _: ^2 y) ^
from matplotlib import pyplot as plt
! G7 }) O$ P/ W9 s4 v
import sklearn
6 E& o1 ^% U2 P; [) d, g
2 W I7 j5 t: a$ c0 a
X_train = np.array([ # 身高体重
) B+ r& f( q9 e I" o
[158, 64],
: t: L5 ?4 h$ ?. K' _' ?1 U* W/ ?6 B1 \
[170, 86],
9 q+ W+ q5 E& }. y
[183, 84],
. Q* r1 }) K! B1 |) M
[191, 80],
4 p4 H1 Y- f9 K
[155, 49],
! u2 ~0 \9 X) r* x; {( R+ K
[163, 59],
$ w2 y1 z9 N% o+ M" r# z9 o- T X
[180, 67],
$ x" M" [" J+ {6 H
[158, 54],
, X7 B9 P& \/ o2 M; `. I9 G
[170, 67]])
+ ^+ ~( B; O. J5 c, {, I5 v
y_train = ['male']*4 + ['female']*5 # 性别
$ H6 n9 N' E! O( s: z4 S
4 t1 L7 ^; B, \8 A4 O: {
#绘制图像
% w9 y+ t' z+ d' b+ i* T% [1 d
plt.figure()
9 B/ M" {3 I z7 R5 q
plt.title('Human Height and Weights by Sex')
! D5 W: v" Z Y% [: a
plt.xlabel('Height in cm')
. { Q6 G: C" n) }9 D3 \3 ]
plt.ylabel('Weight in kg')
1 g+ Y$ I2 W1 ~- q$ S- Q/ D N
for i, x in enumerate(X_train):
8 M$ e4 ?! W+ l- n, V1 B
plt.scatter(x[0],x[1],c='k',marker='x' if y_train
== 'male' else 'D')
3 \8 n) d- U m; B* v. {
plt.grid()
' Z. n0 f0 T' I$ w
plt.show()
, x3 B* O+ h/ s9 C& D
- {; D, _- B$ T: f8 G% c
结果：
, i R2 P( d6 q
" x2 A6 j7 X- x
% c8 M W7 V; h3 y
8 o C1 {0 p& C
我们使用欧几里得距离公式来衡量距离：
* o4 Y2 V4 O8 y8 Z2 {/ @5 N. i
9 X, C( X) a/ L* y8 j: e
' X9 ] k& t8 j9 c6 s
: p3 D x2 u2 S8 k9 ^
+ `# t; \5 o$ o3 q E. N( A6 N0 P
, y; R+ Z- k9 A, M# z+ c4 C
我们设置参数K=3，来寻找3个距离最近的训练实例
; P* _( u3 e- s0 B! i+ b6 c; n
5 h# @7 F# S( z
下面代码实现K近邻算法进行分类：
! {, ~/ J" _) I9 C$ G/ |
$ S& q" a% g: H9 V
x = np.array([[155, 70]])
" E; @; z# h. [
distances = np.sqrt(np.sum((X_train - x)**2, axis=1)) # 计算距离
' k( S5 i; C/ ]: h: w
# @% q" @0 a3 s$ x) H3 {* i
nearest_neighbor_indices = distances.argsort()[:3] # 找出前三个距离最小的下标
& q5 H0 E( E/ h/ n; }4 S4 c7 s
nearest_neighbor_genders = np.take(y_train, nearest_neighbor_indices) # 得到下标对应的标签
6 J% \& i6 ?+ \/ r* u
. J( w5 z" v5 `: O9 {
from collections import Counter
4 d4 G( R3 F# a3 [) Q* n
b = Counter(np.take(y_train, distances.argsort()[:3])) #得到三个结果标签中最频繁的标签得到结果female
F, E" s# |) I
# {$ ?; L1 k4 Y& t
print(b.most_common(1)[0][0]) # female
5 X% ]$ o& M1 |) E0 @+ V
因此，从上述代码可以得到K近邻算法进行分类就是找到离样本点最近的K个实例，再取K个实例的标签中出现次数最多的那个作为我们的结果。
' |6 w; E( N3 b: g0 O0 Y3 P
: \/ u5 T% K" s0 ]- i
上述K近邻算法在scikit-learn中也有对应的函数：
8 ~% G" i, i' i) }" I$ X
, u) n. D5 m- I% Z6 E/ A; k8 c
from sklearn.preprocessing import LabelBinarizer
8 M- T. s { _
from sklearn.neighbors import KNeighborsClassifier
0 c/ h& n8 J' k. A( L
4 j: y0 W% M% B0 y
lb = LabelBinarizer() # 创建将标签二值数值化的类实例
1 {! z! L% o0 {8 {' G D
y_train_binarized = lb.fit_transform(y_train) # 将标签二值数值化
# W7 q- P2 D( L; t) x# F
print(y_train_binarized)
9 r, @3 O% ?3 V
! K" D* a% I$ d* |; d; u& f8 v
K = 3
. W" y; m/ W9 y. y _/ K
clf = KNeighborsClassifier(n_neighbors=K) # 创建K近邻分类器实例
. i% K# A4 s8 z, `- J; c- g
clf.fit(X_train, y_train_binarized.reshape(-1, 1)) # 对训练集进行训练
& |0 I$ P7 B+ a# k
prediction_binarized = clf.predict(np.array([155, 70]).reshape(1,-1))[0] # 对测试样本点进行预测
$ C2 l- P, B& ?8 K1 [* X& _) d
prediction_label = lb.inverse_transform(prediction_binarized) # 将预测结果从数字转换为标签
! N6 A5 e3 s' c7 w5 S; o, D
print(prediction_label) # array(['female'], dtype='<U6')
1 N" A) E" F* T$ d! N: R
KNN回归：
* k3 y# O4 {# p& F) N( p
K近邻算法进行回归和K近邻思想一致，只不过在得到了K个邻居后，分类是取邻居中出现次数最多的那个，而回归是取其他的操作来预测输出值(比如取平均)
! v9 t. @5 f+ E" w+ l% y7 ^2 l
* [; c& x. A+ `# F
对应的代码在scikit-learn中其实也很简单
/ I) Z, e$ Y" D; D# p+ Z+ N2 w# M2 O, H
" E1 X- E+ u- d5 x- t
from sklearn.neighbors import KNeighborsRegressor
# @- Q8 K- u/ _) C J8 i1 s& I
K = 3
9 O, Z7 g, i# C+ \9 U3 i
clf = KNeighborsRegressor(n_neighbors=K)
& t4 X, [6 n3 l; q8 h2 T
clf.fit(X_train, y_train)
4 c/ y8 x. F( t2 ]2 N
predictions = clf.predict(X_test)
/ w) Z6 n0 z% R
特征缩放
N! x' a* V! c/ E! J
下面我们谈谈一个提升算法精确度的小细节。假设还是上面的数据，我们现在要做回归，给定身高和性别标签来预测体重。如果我们的训练数据集包含一个身高170cm的男性和身高160cm的女性。如果我们的测试集数据为身高为164cm的男性，你觉得其预测结果会接近170cm的男性还是身高160cm的女性呢？我们可能相信测试实例更接近男性实例，因为对预测体重来说，性别差异可能会比 6cm 的身高差距更重要。但是如果我们以毫米为单位表示身高，测试实例更接近于身高1600mm 的女性。如果我们以米为单位表示身高，测试实例更接近于身高 1.7m 的男性。（记住我们以欧几里得距离来衡量）
+ m6 Z, q- P& J: S q
1 k2 F/ I$ ]' y- J8 s6 S9 S& s
因此，我们的特征缩放的作用就出来了(其实就相当于深度学习对数据集预处理中的Normalize)
6 m) v5 g# R3 L+ z
# |5 p u3 \+ ?' _% W
将所有实例特征值减去均值来将其居中。其次将每个实例特征值除以特征的标准差对其进行缩放。均值为 0，方差为 1 的数据称为标准化数据。
. g- w8 `8 p+ v2 D1 x( Z2 }
————————————————
0 b, ~2 z3 f" s: ]6 d
版权声明：本文为CSDN博主「王大队长」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
8 i( B# W1 F% L! k" x) a( o
原文链接：https://blog.csdn.net/qq_55621259/article/details/126695549
) r7 g1 T" ]$ R+ [, r# t1 F
/ z S; r5 s. R1 J
' M4 j1 c0 n9 `
欢迎光临 数学建模社区-数学中国 (http://www.madio.net/)
Powered by Discuz! X2.5