分类:首先将测试向量inX(1*dims)与训练样本数据dataSet(n*dims)中的每个向量进行余弦距离计算,然后将结果排序,输出topk个最大值;相对这k个lable进行投票累加,累加值最大的就是分类结果;
python对矩阵运算能力比较强大,所以dims维单向量特征的测试向量与训练data的余弦距离是通过矩阵得到的
其中tile(inX, (dataSetSize,1)),将inX转为n*dims矩阵,其中每一列数据就是inX元素….然后一次相减,平方和,开根号得到的sqDistances 就是测试向量与训练data中n个向量的余弦距离
classCount是python中的字典数据类型,(key-value)
sortedClassCount 是通过value值从大到小排序得到的排号序的字典类型
from numpy import *
import operator
from os import listdir
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
sortedDistIndicies = distances.argsort()
classCount={}
for i in range(k):
voteIlabel = labels[sortedDistIndicies[i]]
classCount[voteIlabel] = classCount.get(voteIlabel,0) + 1
sortedClassCount = sorted(classCount.iteritems(), key=operator.itemgetter(1), reverse=True)
return sortedClassCount[0][0]