本文實(shí)例講述了kNN算法python實(shí)現(xiàn)和簡單數(shù)字識(shí)別的方法。分享給大家供大家參考。具體如下:
kNN算法算法優(yōu)缺點(diǎn):
優(yōu)點(diǎn):精度高、對(duì)異常值不敏感、無輸入數(shù)據(jù)假定
缺點(diǎn):時(shí)間復(fù)雜度和空間復(fù)雜度都很高
適用數(shù)據(jù)范圍:數(shù)值型和標(biāo)稱型
算法的思路:
KNN算法(全稱K最近鄰算法),算法的思想很簡單,簡單的說就是物以類聚,也就是說我們從一堆已知的訓(xùn)練集中找出k個(gè)與目標(biāo)最靠近的,然后看他們中最多的分類是哪個(gè),就以這個(gè)為依據(jù)分類。
函數(shù)解析:
庫函數(shù):
tile()
如tile(A,n)就是將A重復(fù)n次
代碼如下:a = np.array([0, 1, 2])
np.tile(a, 2)
array([0, 1, 2, 0, 1, 2])
np.tile(a, (2, 2))
array([[0, 1, 2, 0, 1, 2],[0, 1, 2, 0, 1, 2]])
np.tile(a, (2, 1, 2))
array([[[0, 1, 2, 0, 1, 2]],[[0, 1, 2, 0, 1, 2]]])
b = np.array([[1, 2], [3, 4]])
np.tile(b, 2)
array([[1, 2, 1, 2],[3, 4, 3, 4]])
np.tile(b, (2, 1))
array([[1, 2],[3, 4],[1, 2],[3, 4]])`
自己實(shí)現(xiàn)的函數(shù)
createDataSet()生成測試數(shù)組
kNNclassify(inputX, dataSet, labels, k)分類函數(shù)
inputX 輸入的參數(shù)
dataSet 訓(xùn)練集
labels 訓(xùn)練集的標(biāo)號(hào)
k 最近鄰的數(shù)目
代碼如下:
#coding=utf-8
from numpy import *
import operator
def createDataSet():
group = array([[1.0, 0.9], [1.0, 1.0], [0.1, 0.2], [0.0, 0.1]])
labels = ['A','A','B','B']
return group,labels
#inputX表示輸入向量(也就是我們要判斷它屬于哪一類的)
#dataSet表示訓(xùn)練樣本
#label表示訓(xùn)練樣本的標(biāo)簽
#k是最近鄰的參數(shù),選最近k個(gè)
def kNNclassify(inputX, dataSet, labels, k):
dataSetSize = dataSet.shape[0]#計(jì)算有幾個(gè)訓(xùn)練數(shù)據(jù)
#開始計(jì)算歐幾里得距離
diffMat = tile(inputX, (dataSetSize,1)) - dataSet
sqDiffMat = diffMat ** 2
sqDistances = sqDiffMat.sum(axis=1)#矩陣每一行向量相加
distances = sqDistances ** 0.5
#歐幾里得距離計(jì)算完畢
sortedDistance = distances.argsort()
classCount = {}
for i in xrange(k):
voteLabel = labels[sortedDistance[i]]
classCount[voteLabel] = classCount.get(voteLabel,0) + 1
res = max(classCount)
return res
def main():
group,labels = createDataSet()
t = kNNclassify([0,0],group,labels,3)
print t
if __name__=='__main__':
新聞熱點(diǎn)
疑難解答
圖片精選