kNN

作者: 空山新雨后丶 | 来源:发表于2017-10-24 23:25 被阅读0次
from numpy import *
import operator

def createDateSet():
    group = array([[1.0, 1.1], [1.0, 1.0], [0, 0], [0, 0.1]])
    labels = ['A', 'A', 'B', 'B']
    return group, labels

def classify0(inX, dataSet, labels, k):
    dataSetSize = dataSet.shape[0]
    diffMat = tile(inX, (dataSetSize, 1)) - dataSet
    sqDiffMat = diffMat**2
    sqDistances = sqDiffMat.sum(axis=1)
    distances = sqDistances**0.5
    sortedDistancesIndicies = distances.argsort()
    classCount = {}
    for i in range(k):
        voteLabel = labels[sortedDistancesIndicies[i]]
        classCount[voteLabel] = classCount.get(voteLabel, 0) + 1
    sortedClassCount = sorted(classCount.items(), key=operator.itemgetter(1), reverse=True)
    return sortedClassCount[0][0]

createDateSet 用来产生已知类别数据集,group是二维数组,是属性(特征);labels是标签,行与行对应,如[1.0, 1.1] 对应着 'A'。

classify0函数接收四个参数,inX是待测试数据,dataSet是已知数据属性,labels是已知数据对应的标签,k是用于选择最近邻居的数目。
shape函数是得到矩阵的尺寸,第0维代表矩阵的行,即我们这边已知数据的组数。
tile函数是把一个一维数组扩展,这里扩三行,列数不变,这里经过扩展之后大小和已知数据集大小一样,都是4×2的矩阵,相减得到差矩阵。
差矩阵平方,**2
平方之后再求和,sum函数的参数axis=1表示按行相加得到一列数。
开根号,**0.5
argsort函数是得到排序序号,最小的序号为0.
定义一个classCount字典。
保留距离最小的k个标签,遍历字典,在字典中保存标签出现的次数。get函数在字典中查找关键字,没找到则返回0,找到关键字(已有的次数)则+1。

把字典排序,sorted接收3个参数,第一个为迭代器,第二个为排序依据,operator.itemgetter函数 是得到迭代器维数的作用,第一维表示得到值,第三个reverse=True表示按照降序排列。返回值为元组类型。

>>> l = {'B': 2, 'A': 1}
>>> ll = sorted(l.items(), key=operator.itemgetter(1), reverse=False)
>>> ll
[('A', 1), ('B', 2)]

>>> fun = operator.itemgetter(1)
>>> fun(ll)
('B', 2)

>>> l.items()
dict_items([('B', 2), ('A', 1)])

返回出现次数最多的标签。

相关文章

  • Spark --基于DataFrame API实现KNN算法

    Spark -- 基于DataFrame API实现KNN算法 KNN简介 KNN(k-Nearest Neigh...

  • KNN近邻算法总结

    目录 一、KNN近邻算法思想 二、KNN模型三大要素 三、KNN算法实现步骤 四、KNN算法的KD树实现 五、总结...

  • 利用Python进行数字识别

    思路 通过Python实现KNN算法。而KNN算法就是K最近邻(k-Nearest Neighbor,KNN)分类...

  • KNN

    KNN学习笔记 KNN is a classification algorithm which is instan...

  • 第六节分类算法

    1knn算法 1.1knn的过程 1.2scilit-learn中的knn 1.3scikit-learn机器学习...

  • knn算法

    knn算法 knn算法简介 邻近算法,或者说K最近邻(kNN,k-NearestNeighbor)分类算法。所谓K...

  • 机器学习笔记汇总

    kNN算法:K最近邻(kNN,k-NearestNeighbor)分类算法

  • 01 KNN算法 - 概述

    KNN算法全称是K近邻算法 (K-nearst neighbors,KNN) KNN是一种基本的机器学习算法,所谓...

  • KNN算法-1-KNN简介

    KNN入门 1、KNN简介 kNN(k-NearestNeighbor),也就是k最近邻算法,这是一种有监督的学习...

  • KNN算法以及欧式距离

    1.KNN算法介绍 KNN 是什么? KNN(K-Nearest Neighbor)是最简单的机器学习算法之一,可...

网友评论

      本文标题:kNN

      本文链接:https://www.haomeiwen.com/subject/xftfpxtx.html