虽然已经熟悉掌握了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