. F8 Y0 S b6 k3 p: B K-近邻算法分类和回归 W- @8 ^& Y, A! y+ S& P" S
K近邻算法的主要思想是用离测试集数据点最近的训练集点(称为其邻居)的输出来估计测试集数据点的输出,参数K代表用多少个邻居来估计。超参数K通常设置为奇数来防止平局现象。3 Q% v k* g! w
( i, `! c+ v5 h其中对邻居的判定:我们可以用欧几里得距离来衡量距离来确定其K个邻居。 F4 m$ [( i# |( a: f+ `7 i
4 u0 S! ~. Q) M, u; gK近邻算法是一种惰性学习和非参数模型。当训练数据数量庞大,同时你对响应变量和解释变量之间的关系所知甚少时,非参数模型会非常有用。KNN 模型只基于一个假设:互相接近的实例拥有类似的响应变量值。非参数模型提供的灵活性并不总是可取的,当训练数据很缺乏或者你对响应变量和解释变量之间的关系有所了解时,对响应变量和解释变量之间关系做假设的模型就很有用。0 I! R6 C8 O8 G/ L; T" i+ i1 b
. a* ]0 M2 b/ c0 DKNN模型分类: 1 m: {" ?$ n7 y- n) K下面我们看一个分类的例子和代码实现来了解一下K近邻算法: - h7 O: o* j {2 }1 h+ }' @( z . ~1 d+ ^: @9 A+ ]+ r8 \ H3 l( i$ h: v
: y3 ^1 J3 L' x* z' @4 l
上表是我们的训练集,下面先对数据进行可视化 / U+ d; i0 F, T9 K; G* Q# J$ d! }% @- g
import numpy as np * [3 a( ]0 ~3 Y- a* L. lfrom matplotlib import pyplot as plt + f0 x& [0 M% A9 |7 c$ l' t- simport sklearn + Y) _5 K- j: N* G & a, F( }0 i- z, {. i6 U! e, BX_train = np.array([ # 身高体重 d9 F" `# {; ]
[158, 64],! K3 Z' E; o) }' J" Z1 d6 m
[170, 86], - S/ ^" A1 l5 X* R& I: A( k0 G [183, 84],/ g7 @ O4 @" g) B: x0 Q( o3 n
[191, 80], ) m, x( F2 c' _5 `2 O [155, 49],. M+ [, b; R+ P2 A2 K9 v1 g. {
[163, 59], 7 H# t! z& ]& Z+ ~. r g& U. w* n [180, 67], # k6 F' [, I$ v' o2 [$ r [158, 54], ' l' ?$ m0 }$ s9 K( `9 P/ K [170, 67]])7 t. ~7 O! G+ _& b) B4 ?) A/ z
y_train = ['male']*4 + ['female']*5 # 性别/ [# q3 ?. j/ T& [% C. A; N
% @1 I' m a2 `3 [$ b2 g) d
#绘制图像/ B x* J" }8 C
plt.figure() # g9 x: {5 h8 Z8 [plt.title('Human Height and Weights by Sex') + p( z6 x) X6 U! ~" dplt.xlabel('Height in cm'); U$ p% G0 m8 j; S7 ]; Z4 a
plt.ylabel('Weight in kg') ) _8 u6 E- t9 E: f% E( q) T/ w. gfor i, x in enumerate(X_train): $ M+ n" `% o" D5 z2 b5 a: A plt.scatter(x[0],x[1],c='k',marker='x' if y_train == 'male' else 'D')1 Y' N) G" n4 k2 g; @6 z9 p
plt.grid()3 R7 \4 J# h0 K( O6 e, V
plt.show()4 @6 P+ p$ z9 t3 k, y( N
( q0 @* l) ^+ g3 J; i5 a
结果: " o# M9 }- Q8 N, w" \" M D$ i ; T5 r7 W: z' T: g3 ^ ~ d! u/ l2 b8 p7 B; L- A$ s( c K8 q
我们使用欧几里得距离公式来衡量距离: . A$ g( m! j) n5 b/ ]$ E5 | ( ~6 s' v7 j0 x K, \ M1 L9 V% `; Y- m6 b
C: |9 M f, i2 K% b* g/ W) u
/ a- g+ ?. G0 S( `' l
: q7 |1 g8 t) }' { w) k, ] 我们设置参数K=3,来寻找3个距离最近的训练实例1 O5 b% A. f; @! s/ {& H8 q7 D! S
/ ^. M- @# ^2 x g. n, b
下面代码实现K近邻算法进行分类:" O8 n6 o/ k$ B3 z