利用python实现KNN分类器

VIP文章 bigface1234fdfg

于 2014-09-27 17:24:44 发布

阅读量2.2k

点赞数 1

分类专栏： Machine Learning&&Data Mining

本文链接：https://blog.csdn.net/puqutogether/article/details/39613257

版权

虽然已经熟悉掌握了KNN分类器，但是关键还是要实现一遍，参考一些资料之后，实现如下代码（加了注释）：

#该程序实现了kNN算法
import numpy
import operator

def kNNClassifier(inX, dataSet, labels, k):
    '''
    inX: 被分类的测试样本（向量形式）；
    dataSet: 训练样本集，每个样本为一行，则行数=样本个数
    labels：每个样本的标记，list形式
    k：近邻个数
   
    sortedClassCount：返回字典，key是排名，value是标记
    '''
    dataSetSize=dataSet.shape[0]  #shape函数返回行数，也就是训练样本个数
    diffMat=numpy.tile(inX, (dataSetSize, 1))-dataSet   #tile复制测试向量，dataSetSize行，1列，为了相减
    sqDif