【统计学习方法读书笔记与算法实现】2-KNN算法


KNN算法的基本思路:

给定一个训练数据集,对于新的输入实例,在训练数据集中找到与该实例最邻近的k个实例,这k个实例的多数属于某个类,就把这个输入实例分为这个类。

三个核心要素:

k:邻近的实例个数

距离:如何度量新的输入实例与训练集中样本的距离【如何度量两个样本点的相似程度】

对于n维实数向量空间Rn,使用欧氏距离;

其他的距离/相似度度量方法有:

http://www.cnblogs.com/daniel-D/p/3244718.html

分类决策规则:多数投票等方法实现分类


KNN算法的实现逻辑:


KNN方法的3要素:

1. 距离:Lp距离

2. k的选择:

如果k=1,则近似误差比较小,但是模型整体变得复杂,容易出现过拟合;

如果k=N(样本数),则近似误差大,模型整体变得简单(预测值都是训练集中出现次数最多的类别),但是忽略了训练实例中的大量有效信息

采用交叉验证的方法来选择k

3. 分类决策规则:

多数表决规则

K邻近算法的实现:kd树

问题:如何对训练数据进行快速的k邻近搜索;在特征空间维数大及训练数据容量大时尤其必要


KNN算法的简单实现:

import numpy as np

import operator

# 训练集

data_set = np.array([[1., 1.1],

[1.0, 1.0],

[0., 0.],

[0, 0.1]])

labels = ['A', 'A', 'B', 'B']

def classify_knn(in_vector,training_data,training_label,k):

"""

:param in_vector:待分类向量

:param training_data:训练集向量

:param training_label:训练集标签

:param k:选择最近邻居的数目

:return:分类器对 in_vector 分类的类别

"""

data_size=training_data.shape[0] # 若data为m行n列的二维矩阵,则data.shape=[m,n], data.shape[0]=m,也就是矩阵的行数

diff_mat=np.tile(in_vector,(data_size, 1))-data_set

sq_diff_mat=diff_mat**2 #array的**指对array中的每个元素求平方

sq_distances=sq_diff_mat.sum(axis=1) #numpy中的sum函数,若没有axis表示所有元素相加,axis=0表示按列相加,axis=1表示按行相加

distance_sorted_index=sq_distances.argsort() #numpy中的argsort()函数,返回sq_distance矩阵中的列元素,按照从小到大的顺序排列得到的索引值

class_count_dict = {}  # 用于统计类别的个数

for i in range(k):

label = training_label[distance_sorted_index[i]]

try:

class_count_dict[label] += 1

except KeyError:

class_count_dict[label] = 1

class_count_dict = sorted(class_count_dict.items(), key=operator.itemgetter(1), reverse=True)

# operator.itemgetter(1)表示定义了函数key,获取对象的第一个域的值

# class_count_dict.items()表示返回class_count_dict中可遍历的(键,值)元组数组

return class_count_dict[0][0]

if __name__ == '__main__':

vector = [0, 0.3]  # 待分类数据集

print(classify_knn(in_vector=vector, training_data=data_set, training_label=labels, k=3))


参考资料:

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 157,198评论 4 359
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 66,663评论 1 290
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 106,985评论 0 237
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,673评论 0 202
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 51,994评论 3 285
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,399评论 1 211
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,717评论 2 310
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,407评论 0 194
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,112评论 1 239
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,371评论 2 241
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 31,891评论 1 256
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,255评论 2 250
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,881评论 3 233
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,010评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,764评论 0 192
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,412评论 2 269
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,299评论 2 260

推荐阅读更多精彩内容