K近邻算法及其Python实现

本文转载自查看原文 2016-07-12 17:02 2607 机器学习/ Python

下面贴出Python代码

knnClassify.py

 1 from numpy import *
 2 import operator
 3 
 4 def creatDataSet():
 5     group = array([[1.0,1.1],[1.0,1.0],[0,0],[0,0.1]])
 6     labels = ['A','A','B','B']
 7     return group, labels
 8 
 9 def classify(inX,dataSet,labels,k):
10     numSamples = dataSet.shape[0]
11     diffMat = tile(inX,(numSamples,1)) - dataSet
12     sqDiffMat = diffMat**2
13     sqDistances = sqDiffMat.sum(axis = 1)
14     distances = sqDistances ** 0.5
15     sortedDistIndicies = distances.argsort()
16     classCount = {}
17     for i in xrange(k):
18         voteILabel = labels[sortedDistIndicies[i]]
19         classCount[voteILabel] = classCount.get(voteILabel,0) + 1
20     maxLabel = sorted(classCount.iteritems(),
21                       key = operator.itemgetter(1),reverse = True)
22     return maxLabel[0][0]
23 
24 if __name__=="__main__":
25     g,l = creatDataSet()
26     labelOfinX = classify([1.0,1.2],g,l,1)
27     print labelOfinX

免责声明！

本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。

猜您在找 k近邻算法的Python实现用python实现k近邻算法 k-近邻算法 python实现 Python 实现 KNN（K-近邻）算法 k近邻算法-java实现 k近邻算法的Java实现 K-近邻算法的Python实现：源代码分析机器学习三 -- 用Python实现K-近邻算法 k-近邻算法python代码实现（非常全） KNN算法(K近邻算法)实现与剖析