Machine Learn in Action(K-近鄰算法)
阿新 • • 發佈:2017-06-27
count rom sha group .get name imp diff mac
使用K-近鄰算法將某點[0.6, 0.6]劃分到某個類(A, B)中。
from numpy import * import operator 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 # operator.itemgetter(1)根據iterable的第二個值域排序 sortedClassCount = sorted(classCount.items(), key=operator.itemgetter(1), reverse=True)return sortedClassCount[0][0] if __name__ == ‘__main__‘: # 定義訓練集 group = array([[1.0, 1.1], [1.0, 1.0], [0, 0], [0, 0.1]]) labels = [‘A‘, ‘A‘, ‘B‘, ‘B‘] print(classify0([0.6, 0.6], group, labels, 3))
Machine Learn in Action(K-近鄰算法)