2-1節(jié) k-近鄰算法-KNN算法|優(yōu)化約會網站的配對效果項目|機器學習實戰(zhàn)-學習筆記

文章原創(chuàng),最近更新:2018-08-7

本章節(jié)的主要內容是:
重點介紹項目案例1: 優(yōu)化約會網站的配對效果中的 KNN算法。

1.KNN項目案例介紹:

項目案例1:

優(yōu)化約會網站的配對效果

項目概述:

1)海倫使用約會網站尋找約會對象。經過一段時間之后,她發(fā)現曾交往過三種類型的人: 不喜歡的人、魅力一般的人、 極具魅力的人。
2)她希望: 1. 工作日與魅力一般的人約會 2. 周末與極具魅力的人約會 3. 不喜歡的人則直接排除掉。現在她收集到了一些約會網站未曾記錄的數據信息,這更有助于匹配對象的歸類。

開發(fā)流程:
  • 收集數據:提供文本文件
  • 準備數據:使用 Python 解析文本文件
  • 分析數據:使用 Matplotlib 畫二維散點圖
  • 訓練算法:此步驟不適用于 k-近鄰算法
  • 測試算法:使用海倫提供的部分數據作為測試樣本。
    測試樣本和非測試樣本的區(qū)別在于:測試樣本是已經完成分類的數據,如果預測分類與實際類別不同,則標記為一個錯誤。
  • 使用算法:產生簡單的命令行程序,然后海倫可以輸入一些特征數據以判斷對方是否為自己喜歡的類型。
數據集介紹

海倫把這些約會對象的數據存放在文本文件 datingTestSet2.txt (數據來源于《機器學習實戰(zhàn)》第二章 k鄰近算法)中,總共有 1000 行。

本文使用的數據主要包含以下三種特征:每年獲得的飛行常客里程數,玩視頻游戲所耗時間百分比,每周消費的冰淇淋公升數。其中分類結果作為文件的第四列,并且只有3、2、1三種分類值。datingTestSet2.csv文件格式如下所示:

飛行里程數 游戲耗時百分比 冰淇淋公升數 分類結果
40920 8.326976 0.953952 3
14488 7.153469 1.673904 2
26052 1.441871 0.805124 1

數據在datingTestSet2.txt文件中的格式如下所示:


2.KNN算法代碼

首先創(chuàng)建一個名為kNN.py的文件,再創(chuàng)建一個函數,這個函數返回一個矩陣和標簽列表,以方便我們后續(xù)對K近鄰(KNN)算法進行檢驗。

def createDataSet():
    """
    創(chuàng)建數據集和標簽

     調用方式
     import kNN
     group, labels = kNN.createDataSet()
    """
    group = array([[1.0, 1.1], [1.0, 1.0], [0, 0], [0, 0.1]])
    labels = ['A', 'A', 'B', 'B']
    return group, labels

下面這段代碼就是kNN算法,目的就是為了找最近的距離

def classify0(inX,dataSet,labels,k):
    """
    inX:用于分類的輸入向量
    dataSet:輸入的訓練樣本集
    lables:標簽向量
    k:表示用于選擇最近鄰居的數目
    
    預測數據所在分類可在輸入下列命令
    kNN.classify0([0,0], group, labels, 3)
    """
    # array的shape函數返回指定維度的大小,如dataset為n*m的矩陣,
    # 則dataset.shape[0]返回n,dataset.shape[1]返回m,dataset.shape返回n,m
    dataSetSize = dataSet.shape[0]
    # tile函數簡單的理解,它的功能是重復某個數組。比如tile(A,n),功能是將數組A重復n次,構成一個新的數組
    # 所以此處tile(inX,(dataSetSize,1))的作用是將inX重復復制dataSetSize次,以便與訓練樣本集的樣本個數一致
    # 減去dataSet就是求出其差值,所以diffMat為一個差值矩陣
    diffmat=np.tile(inX,(dataSetSize,1))-dataSet
    #距離度量,度量公式為歐氏距離
    sqdiffmat=diffmat**2
    # 將矩陣的每一行相加,axis用于控制是行相加還是列相加
    sqdistances=sqdiffmat.sum(axis=1)
    #開方
    distances=sqdistances**0.5
    # 根據距離排序從小到大的排序,返回對應的索引位置
    sortedDistIndicies=distances.argsort()
    # 選擇距離最小的k個點
    classcount={}
   
    for i in range(k):
        # 找到該樣本標簽的類型
        voteIlabel=labels[sortedDistIndicies[i]]
        # 字典的get方法,list.get(k,d) 其中 get相當于一條if...else...語句,參數k在字典中,字典將返回list[k];如果參數k不在字典中則返回參數d
        classcount[voteIlabel]=classcount.get(voteIlabel,0)+1
        # 字典的 items() 方法,以列表返回可遍歷的(鍵,值)元組數組。
        # sorted 中的第2個參數 key=operator.itemgetter(1) 這個參數的意思是先比較第幾個元素
        sortedClasscount = sorted(classcount.items(),key=operator.itemgetter(1),reverse=True)
        # 返回最符合的標簽
        return sortedClasscount[0][0]

測試代碼及其結果如下:

>import kNN
>group,labels=kNN.createDataSet()
>kNN.classify0([0,0],group,labels,3)

'B'

3.KNN算法相關知識點

知識點1:歐氏距離

計算兩個向量點xA和xB之間的距離,叫做歐氏距離公式,具體如下:
d=\sqrt{(xA_{0}-xB_{0})^{2}+(xA_{1}-xB_{1})^{2}}
例如,點(0,0)與(1,2)之間的距離計算為:
\sqrt{(0-1)^{2}+(1-2)^{2}}
舉個小案例,首先求點與點之間的距離,具體如下

  • 第一行: 同一個點 到 dataSet的第一個點的距離。
  • 第二行: 同一個點 到 dataSet的第二個點的距離。
  • ...
  • 第N行: 同一個點 到 dataSet的第N個點的距離。

    由此可以得到截圖的歐氏距離是:
    a_{1}=\sqrt{(x-1)^{2}+(y-1)^{2}}
    a_{2}=\sqrt{(x-1.1)^{2}+(y-1)^{2}}
    a_{3}=\sqrt{(x-0)^{2}+(y-0)^{2}}
    a_{4}=\sqrt{(x-0)^{2}+(y-0.1)^{2}}

知識點2:關于operator模塊的運用

operator模塊是python中內置的操作符函數接口,它定義了一些算術和比較內置操作的函數。operator模塊是用c實現的,所以執(zhí)行速度比python代碼快。

知識點3:numpy中c.shape[0]的理解

建立一個4×2的矩陣c, c.shape[0] 為第一維的長度4(相當于行),c.shape[1] 為第二維的長度2(相當于列)。

import numpy as np
group = np.array([[1.0,1.1],[1.0,1.0],[0,0],[0,0.1]])

group
Out[21]: 
array([[ 1. ,  1.1],
       [ 1. ,  1. ],
       [ 0. ,  0. ],
       [ 0. ,  0.1]])

group.shape
Out[22]: (4, 2)

group.shape[0]
Out[23]: 4

group.shape[1]
Out[24]: 2

知識點4:numpy中的tile函數

numpy.tile(A,B)函數,作用使數組A重復B次,這里的B可以時int類型也可以是元組類型。

import numpy as np

np.tile([0,1],2)#在列方向上重復[0,1]2次,默認行1次
Out[26]: array([0, 1, 0, 1])

np.tile([0,1],(1,1))#在列方向上重復[0,1]1次,行1次
Out[27]: array([[0, 1]])

np.tile([0,1],(2,1))#在列方向上重復[0,1]1次,行2次
Out[28]: 
array([[0, 1],
       [0, 1]])

np.tile([0,1],(2,2)))#在列方向上重復[0,1]2次,行2次
Out[29]: 
array([[0, 1, 0, 1],
       [0, 1, 0, 1]])

np.tile([0,1],(3,2)))#在列方向上重復[0,1]2次,行3次
Out[30]: 
array([[0, 1, 0, 1],
       [0, 1, 0, 1],
       [0, 1, 0, 1]])

知識點5:關于diffMat=np.tile(inX,(dataSetSize,1))-dataSet這段代碼的理解

舉了一個小案例,具體可以參見如下案例:

import numpy as np

dataSet = np.array([[1.0,1.1],[1.0,1.0],[0,0],[0,0.1]])

dataSet
Out[47]: 
array([[ 1. ,  1.1],
       [ 1. ,  1. ],
       [ 0. ,  0. ],
       [ 0. ,  0.1]])

inX=np.array([0,1])

inX
Out[49]: array([0, 1])

dataSetSize=dataSet.shape[0]

dataSetSize
Out[51]: 4

tile(inX, (dataSetSize,1))
Out[53]: 
array([[0, 1],
       [0, 1],
       [0, 1],
       [0, 1]])

diffMat=tile(inX, (dataSetSize,1))-group

diffMat
Out[55]: 
array([[-1. , -0.1],
       [-1. ,  0. ],
       [ 0. ,  1. ],
       [ 0. ,  0.9]])

相當于截圖,如下:


知識點6:python中的sum函數.sum(axis=1)

我們平時用的sum應該是默認的矩陣所有的值相加;axis=1表示按列求和,即把每一行的元素加起來;axis=0表示按行求和,即把每一列的元素加起來

  • axis = 0代表行相加
  • axis = 1 代表列相加
import numpy as np
a=np.array([[0,2,1],[5,4,2]])

a
Out[59]: 
array([[0, 2, 1],
       [5, 4, 2]])

a.sum()
Out[60]: 14

a.sum(axis=0)
Out[61]: array([5, 6, 3])

a.sum(axis=1)
Out[62]: array([ 3, 11])

知識點7:numpy.argsort函數

返回值為從小到大的數字對應的index, 其中 axis = 0沿著行比較,即第一行和第二行的數比較,axis = 1 為沿著列比較,即第一列和第二列的比較。通過參數sort 還可以選擇排序方法.

a=np.array([5,4,7])

a
Out[64]: array([5, 4, 7])

np.argsort(a)
Out[65]: array([1, 0, 2], dtype=int64)

a
Out[68]: 
array([[1, 5],
       [3, 2]])

np.argsort(a,axis=0)
Out[69]: 
array([[0, 1],
       [1, 0]], dtype=int64)

np.argsort(a,axis=1)
Out[70]: 
array([[0, 1],
       [1, 0]], dtype=int64)

知識點8:classCount = {}

其中{}表示生成的是字典,在字典這個類中,有方法get,對classCount元素賦值,其實是個計數器

知識點9:字典的 get 函數

d.get(k,< default>),表示鍵k存在,則返回相應值,不在則返回< default>值

在這里主要是利用dictionary的get( ) 方法做計數統(tǒng)計,具體小案例如下:
案例1
假設統(tǒng)計 s="aabbccc"中,每個字符出現的次數:
結果應當為:"a":2, "b":2, "c":3

s="aabbccc"
dic={}
for ch in s:
    dic[ch]=1+dic.get(ch,0)
    
print(dic)
{'a': 2, 'b': 2, 'c': 3}

案例2

a={5:2,3:4}

a.get(3,0)
Out[35]: 4

a.get(1,0)
Out[36]: 0

知識點10:字典的items函數

d.items()以列表返回可遍歷的(鍵, 值) 元組數組

dict = {'Name': 'Runoob', 'Age': 7}

print ("Value : %s" %  dict.items())
Value : dict_items([('Name', 'Runoob'), ('Age', 7)])

知識點11:Python sorted() 函數

sorted() 函數對所有可迭代的對象進行排序操作。

具體sorted 語法,如下:

sorted(iterable[, cmp[, key[, reverse]]])

參數說明:

  • iterable -- 可迭代對象。
  • cmp -- 比較的函數,這個具有兩個參數,參數的值都是從可迭代對象中取出,此函數必須遵守的規(guī)則為,大于則返回1,小于則返回-1,等于則返回0。
  • key -- 主要是用來進行比較的元素,只有一個參數,具體的函數的參數就是取自于可迭代對象中,指定可迭代對象中的一個元素來進行排序。
  • reverse -- 排序規(guī)則,reverse = True 降序 , reverse = False 升序(默認)。

具體可參見:菜鳥教程之Python sorted() 函數

具體用法如下:

>>>a = [5,7,6,3,4,1,2]
>>> b = sorted(a)       # 保留原列表
>>> a 
[5, 7, 6, 3, 4, 1, 2]
>>> b
[1, 2, 3, 4, 5, 6, 7]
 
>>> L=[('b',2),('a',1),('c',3),('d',4)]
>>> sorted(L, cmp=lambda x,y:cmp(x[1],y[1]))   # 利用cmp函數
[('a', 1), ('b', 2), ('c', 3), ('d', 4)]
>>> sorted(L, key=lambda x:x[1])               # 利用key
[('a', 1), ('b', 2), ('c', 3), ('d', 4)]
 
 
>>> students = [('john', 'A', 15), ('jane', 'B', 12), ('dave', 'B', 10)]
>>> sorted(students, key=lambda s: s[2])            # 按年齡排序
[('dave', 'B', 10), ('jane', 'B', 12), ('john', 'A', 15)]
 
>>> sorted(students, key=lambda s: s[2], reverse=True)       # 按降序
[('john', 'A', 15), ('jane', 'B', 12), ('dave', 'B', 10)]
>>>

而這里所涉及到的知識點,具體用法可以有以下兩種:
key為函數,指定取待排序元素的哪一項進行排序,函數用上面的例子來說明,代碼如下:
方法一

sorted(classcount.items(),key=lambda classcount.item :classcount.item[1],reverse=True)

key指定的lambda函數功能是去元素classcount.item的第二個域(即:classcount.item[1],),因此sorted排序時,會以classcount.items所有元素的第二個域來進行排序。

方法二
有了下面的operator.itemgetter函數,也可以用該函數來實現,例如要通過student的第三個域排序,可以這么寫:

sorted(classcount.items(),key=operator.itemgetter(1),reverse=True)

知識點12:operator.itemgetter函數

operator模塊提供的itemgetter函數用于獲取對象的哪些維的數據,參數為一些序號(即需要獲取的數據在對象中的序號),下面看例子。
案例1

import operator
a=[1,2,3,4]
b=operator.itemgetter(0)

b(a)
Out[82]: 1

b=operator.itemgetter(1)

b(a)
Out[84]: 2

案例2

a=[('b',2),('a',1),('c',0)]
b=sorted(a,key=operator.itemgetter(1))

b
Out[39]: [('c', 0), ('a', 1), ('b', 2)]
#可以看到排序是按照后邊的0,1,2進行排序的,而不是a,b,c

b=sorted(a,key=operator.itemgetter(0))

b
Out[41]: [('a', 1), ('b', 2), ('c', 0)]
#這次比較的是前邊的a,b,c而不是0,1,2

b=sorted(a,key=operator.itemgetter(1,0))

b
Out[44]: [('c', 0), ('a', 1), ('b', 2)]
#這個是先比較第2個元素,然后對第一個元素進行排序,形成多級排序。
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
【社區(qū)內容提示】社區(qū)部分內容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發(fā)布,文章內容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內容

友情鏈接更多精彩內容