python機器學習實戰(一)
版權聲明:本文為博主原創文章,轉載請指明轉載地址
www.cnblogs.com/fydeblog/p/7140974.html
前言
這篇notebook是關於機器學習中監督學習的k近鄰算法,將介紹2個實例,分別是使用k-近鄰算法改進約會網站的效果和手寫識別系統.
操作系統:ubuntu14.04 運行環境:anaconda-python2.7-notebook 參考書籍:機器學習實戰 notebook writer ----方陽
k-近鄰算法(kNN)的工作原理:存在一個樣本數據集合,也稱作訓練樣本集,並且樣本集中的每個數據都存在標簽,即我們知道樣本集中每一組數據與所屬分類的對應關系,輸入沒有標簽的新數據后,將新數據的每個特征與樣本集中數據對應的特征進行比較,然后算法提取樣本集中特征最相似的分類標簽。
注意事項:在這里說一句,默認環境python2.7的notebook,用python3.6的會出問題,還有我的目錄可能跟你們的不一樣,你們自己跑的時候記得改目錄,我會把notebook和代碼以及數據集放到結尾的百度雲盤,方便你們下載!
1.改進約會網站的匹配效果
1-1.准備導入數據
1 from numpy import * 2 import operator 3 4 def createDataSet(): 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
先來點開胃菜,在上面的代碼中,我們導入了兩個模塊,一個是科學計算包numpy,一個是運算符模塊,在后面都會用到,在createDataSet函數中,我們初始化了group,labels,我們將做這樣一件事,[1.0,1.1]和[1.0,1.0] 對應屬於labels中 A 分類,[0,0]和[0,0.1]對應屬於labels中的B分類,我們想輸入一個新的二維坐標,根據上面的坐標來判斷新的坐標屬於那一類,在這之前,我們要實現k-近鄰算法,下面就開始實現
1 def classify0(inX, dataSet, labels, k): 2 dataSetSize = dataSet.shape[0] 3 diffMat = tile(inX, (dataSetSize,1)) - dataSet 4 sqDiffMat = diffMat**2 5 sqDistances = sqDiffMat.sum(axis=1) 6 distances = sqDistances**0.5 7 sortedDistIndicies = distances.argsort() 8 classCount={} 9 for i in range(k): 10 voteIlabel = labels[sortedDistIndicies[i]] 11 classCount[voteIlabel] = classCount.get(voteIlabel,0) + 1 12 sortedClassCount = sorted(classCount.iteritems(), key=operator.itemgetter(1), reverse=True) 13 return sortedClassCount[0][0]
代碼解析:
函數的第一行是要得到數據集的數目,例如group.shape就是(4,2),shape[0]反應數據集的行,shape[1]反應列數
函數的第二行是array對應相減,tile會生成關於Inx的dataSetSize大小的array,例如,InX是[0,0],則tile(InX,(4,1))是array([[0, 0], [0, 0], [0, 0],[0, 0]]),然后與dataSet對應相減,得到新的array
函數的第三行是對第二步的結果進行平方算法,方便下一步算距離
函數的第四行是進行求和,注意是axis=1,也就是array每個二維數組成員進行求和(行求和),如果是axis=0就是列求和
第五行是進行平方距離的開根號
以上5行實現的是距離的計算 ,下面的是選出距離最小的k個點,對類別進行統計,返回所占數目多的類別
classCount定義為存儲字典,里面有‘A’和‘B’,它們的值是在前k個距離最小的數據集中的個數,本例最后classCount={'A':1,'B':2},函數argsort是返回array數組從小到大的排列的序號,get函數返回字典的鍵值,由於后面加了1,所以每次出現鍵值就加1,就可以就算出鍵值出現的次數里。最后通過sorted函數將classCount字典分解為列表,sorted函數的第二個參數導入了運算符模塊的itemgetter方法,按照第二個元素的次序(即數字)進行排序,由於此處reverse=True,是逆序,所以按照從大到小的次序排列。
1-2.准備數據:從文本中解析數據
這上面是k-近鄰的一個小例子,我的標題還沒介紹,現在來介紹標題,准備數據,一般都是從文本文件中解析數據,還是從一個例子開始吧!
本次例子是改進約會網站的效果,我們定義三個特征來判別三種類型的人
特征一:每年獲得的飛行常客里程數
特征二:玩視頻游戲所耗時間百分比
特征三:每周消費的冰淇淋公升數
根據以上三個特征:來判斷一個人是否是自己不喜歡的人,還是魅力一般的人,還是極具魅力的人
於是,收集了1000個樣本,放在datingTestSet2.txt中,共有1000行,每一行有四列,前三列是特征,后三列是從屬那一類人,於是問題來了,我們這個文本文件的輸入導入到python中來處理,於是需要一個轉換函數file2matrix,函數輸入是文件名字字符串,輸出是訓練樣本矩陣(特征矩陣)和類標簽向量
1 def file2matrix(filename): 2 fr = open(filename) 3 numberOfLines = len(fr.readlines()) #get the number of lines in the file 4 returnMat = zeros((numberOfLines,3)) #prepare matrix to return 5 classLabelVector = [] #prepare labels return 6 fr = open(filename) 7 index = 0 8 for line in fr.readlines(): 9 line = line.strip() 10 listFromLine = line.split('\t') 11 returnMat[index,:] = listFromLine[0:3] 12 classLabelVector.append(int(listFromLine[-1])) 13 index += 1 14 return returnMat,classLabelVector
這個函數比較簡單,就不詳細說明里,這里只介紹以下一些函數的功能吧!
open函數是打開文件,里面必須是字符串,由於后面沒加‘w’,所以是讀文件
readlines函數是一次讀完文件,通過len函數就得到文件的行數
zeros函數是生成numberOfLines X 3的矩陣,是array型的
strip函數是截掉所有的回車符
split函數是以輸入參數為分隔符,輸出分割后的數據,本例是制表鍵,最后輸出元素列表
append函數是向列表中加入數據
1-3.分析數據:使用Matplotlib創建散點圖
首先,從上一步得到訓練樣本矩陣和類標簽向量,先更換一下路徑
cd /home/fangyang/桌面/machinelearninginaction/Ch02/
datingDataMat,datingLabels = file2matrix('datingTestSet2.txt')
1 import matplotlib 2 import matplotlib.pyplot as plt 3 fig = plt.figure() 4 ax = fig.add_subplot(111) 5 ax.scatter(datingDataMat[:,0], datingDataMat[:,1], 15.0*array(datingLabels), 15.0*array(datingLabels)) #scatter函數是用來畫散點圖的 6 plt.show()
結果顯示
1-4. 准備數據: 歸一化處理
我們從上圖可以上出,橫坐標的特征值是遠大於縱坐標的特征值的,這樣再算新數據和數據集的數據的距離時,數字差值最大的屬性對計算結果的影響最大,我們就可能會丟失掉其他屬性,例如這個例子,每年獲取的飛行常客里程數對計算結果的影響遠大於其余兩個特征,這是我們不想看到的,所以這里采用歸一化數值處理,也叫特征縮放,用於將特征縮放到同一個范圍內。
本例的縮放公式 newValue = (oldValue - min) / (max - min)
其中min和max是數據集中的最小特征值和最大特征值。通過該公式可將特征縮放到區間(0,1)
下面是特征縮放的代碼
1 def autoNorm(dataSet): 2 minVals = dataSet.min(0) 3 maxVals = dataSet.max(0) 4 ranges = maxVals - minVals 5 normDataSet = zeros(shape(dataSet)) 6 m = dataSet.shape[0] 7 normDataSet = dataSet - tile(minVals, (m,1)) 8 normDataSet = normDataSet/tile(ranges, (m,1)) #element wise divide 9 return normDataSet, ranges, minVals
normDataSet(1000 X 3)是歸一化后的數據,range(1X3)是特征的范圍差(即最大值減去最小值),minVals(1X3)是最小值。
原理上面已介紹,這里不在復述。
1-5.測試算法:作為完整程序驗證分類器
好了,我們已經有了k-近鄰算法、從文本解析出數據、還有歸一化處理,現在可以使用之前的數據進行測試了,測試代碼如下
1 def datingClassTest(): 2 hoRatio = 0.50 3 datingDataMat,datingLabels = file2matrix('datingTestSet2.txt') #load data setfrom file 4 normMat, ranges, minVals = autoNorm(datingDataMat) 5 m = normMat.shape[0] 6 numTestVecs = int(m*hoRatio) 7 errorCount = 0.0 8 for i in range(numTestVecs): 9 classifierResult = classify0(normMat[i,:],normMat[numTestVecs:m,:],datingLabels[numTestVecs:m],3) 10 print "the classifier came back with: %d, the real answer is: %d" % (classifierResult, datingLabels[i]) 11 if (classifierResult != datingLabels[i]): errorCount += 1.0 12 print "the total error rate is: %f" % (errorCount/float(numTestVecs)) 13 print errorCount
這里函數用到里之前講的三個函數:file2matrix、autoNorm和classify0.這個函數將數據集分成兩個部分,一部分當作分類器的訓練樣本,一部分當作測試樣本,通過hoRatio進行控制,函數hoRatio是0.5,它與樣本總數相乘,將數據集平分,如果想把訓練樣本調大一些,可增大hoRatio,但最好不要超過0.8,以免測試樣本過少,在函數的最后,加了錯誤累加部分,預測出來的結果不等於實際結果,errorCount就加1,然后最后除以總數就得到錯誤的概率。
說了這么多,都還沒有測試以下,下面來測試一下!先從簡單的開始(已將上面的函數放在kNN.py中了)
1 import kNN 2 group , labels = kNN.createDataSet()
group #結果在下
array([[ 1. , 1.1], [ 1. , 1. ], [ 0. , 0. ], [ 0. , 0.1]])
labels #結果在下
['A', 'A', 'B', 'B']
這個小例子最開始提過,有兩個分類A和B,通過上面的group為訓練樣本,測試新的數據屬於那一類
1 kNN.classify0([0,0], group, labels, 3) #使用k-近鄰算法進行測試
'B' #結果是B分類
直觀地可以看出[0,0]是與B所在的樣本更近,下面來測試一下約會網站的匹配效果
先將文本中的數據導出來,由於前面在分析數據畫圖的時候已經用到里file2matrix,這里就不重復用了。
datingDataMat #結果在下
array([[ 4.09200000e+04, 8.32697600e+00, 9.53952000e-01], [ 1.44880000e+04, 7.15346900e+00, 1.67390400e+00], [ 2.60520000e+04, 1.44187100e+00, 8.05124000e-01], ..., [ 2.65750000e+04, 1.06501020e+01, 8.66627000e-01], [ 4.81110000e+04, 9.13452800e+00, 7.28045000e-01], [ 4.37570000e+04, 7.88260100e+00, 1.33244600e+00]])
datingLabels #由於過長,只截取一部分,詳細去看jupyter notebook
然后對數據進行歸一化處理
1 normMat , ranges , minVals = kNN.autoNorm(datingDataMat) #使用歸一化函數
normMat
array([[ 0.44832535, 0.39805139, 0.56233353], [ 0.15873259, 0.34195467, 0.98724416], [ 0.28542943, 0.06892523, 0.47449629], ..., [ 0.29115949, 0.50910294, 0.51079493], [ 0.52711097, 0.43665451, 0.4290048 ], [ 0.47940793, 0.3768091 , 0.78571804]])
ranges
1 kNN.datingClassTest()
由於過長,只截取一部分,詳細去看jupyter notebook
可以看到上面結果出現錯誤32個,錯誤率6.4%,所以這個系統還算不錯!
1-6.系統實現
我們可以看到,測試固然不錯,但用戶交互式很差,所以結合上面,我們要寫一個完整的系統,代碼如下:
1 def classifyPerson(): 2 resultList = ['not at all', 'in small doses', 'in large doses'] 3 percentTats = float(raw_input("percentage of time spent playing video games?")) 4 ffMiles = float(raw_input("frequent flier miles earned per year?")) 5 iceCream = float(raw_input("liters of ice cream consumed per year?")) 6 datingDataMat, datingLabels = file2matrix('datingTestSet2.txt') 7 normMat, ranges, minVals = autoNorm(datingDataMat) 8 inArr = array([ffMiles, percentTats, iceCream]) 9 classifierResult = classify0((inArr-minVals)/ranges, normMat, datingLabels,3) 10 print "You will probably like this person" , resultList[classifierResult - 1]
運行情況
1 kNN.classifyPerson()
percentage of time spent playing video games?10 #這里的數字都是用戶自己輸入的 frequent flier miles earned per year?10000 liters of ice cream consumed per year?0.5 You will probably like this person in small doses
這個就是由用戶自己輸出參數,並判斷出感興趣程度,非常友好
2. 手寫識別系統
下面再介紹一個例子,也是用k-近鄰算法,去實現對一個數字的判斷,首先我們是將寬高是32X32的像素的黑白圖像轉換成文本文件存儲,但我們知道文本文件必須轉換成特征向量,才能進入k-近鄰算法中進行處理,所以我們需要一個img2vector函數去實現這個功能!
img2vector代碼如下:
1 def img2vector(filename): 2 returnVect = zeros((1,1024)) 3 fr = open(filename) 4 for i in range(32): 5 lineStr = fr.readline() 6 for j in range(32): 7 returnVect[0,32*i+j] = int(lineStr[j]) 8 return returnVect
這個函數挺簡單的,先用zeros生成1024的一維array,然后用兩重循環,外循環以行遞進,內循環以列遞進,將32X32的文本數據依次賦值給returnVect
好了,轉換函數寫好了,說一下訓練集和測試集,所有的訓練集都放在trainingDigits文件夾中,測試集放在testDigits文件夾中,訓練集有兩千個樣本,0~9各有200個,測試集大約有900個樣本,這里注意一點,所有在文件夾里的命名方式是有要求的,我們是通過命名方式來解析出它的真實數字,然后與通過k-近鄰算法得出的結果相對比,例如945.txt,這里的數字是9,連接符前面的數字就是這個樣本的真實數據。該系統實現的方法與前面的約會網站的類似,就不多說了。
系統測試代碼如下
1 def handwritingClassTest(): 2 hwLabels = [] 3 trainingFileList = listdir('trainingDigits') #load the training set 4 m = len(trainingFileList) 5 trainingMat = zeros((m,1024)) 6 for i in range(m): 7 fileNameStr = trainingFileList[i] 8 fileStr = fileNameStr.split('.')[0] #take off .txt 9 classNumStr = int(fileStr.split('_')[0]) 10 hwLabels.append(classNumStr) 11 trainingMat[i,:] = img2vector('trainingDigits/%s' % fileNameStr) 12 testFileList = listdir('testDigits') #iterate through the test set 13 errorCount = 0.0 14 mTest = len(testFileList) 15 for i in range(mTest): 16 fileNameStr = testFileList[i] 17 fileStr = fileNameStr.split('.')[0] #take off .txt 18 classNumStr = int(fileStr.split('_')[0]) 19 vectorUnderTest = img2vector('testDigits/%s' % fileNameStr) 20 classifierResult = classify0(vectorUnderTest, trainingMat, hwLabels, 3) 21 print "the classifier came back with: %d, the real answer is: %d" % (classifierResult, classNumStr) 22 if (classifierResult != classNumStr): errorCount += 1.0 23 print "\nthe total number of errors is: %d" % errorCount 24 print "\nthe total error rate is: %f" % (errorCount/float(mTest))
這里的listdir是從os模塊導入的,它的功能是列出給定目錄下的所有文件名,以字符串形式存放,輸出是一個列表
這里的split函數是要分離符號,得到該文本的真實數據,第一個split函數是以小數點為分隔符,例如‘1_186.txt’ ,就變成了['1_186','txt'],然后取出第一個,就截掉了.txt,第二個split函數是以連接符_為分隔符,就截掉后面的序號,剩下前面的字符數據‘1’,然后轉成int型就得到了它的真實數據,其他的沒什么,跟前面一樣
下面開始測試
1 kNN.handwritingClassTest()
我們可以看到最后結果,錯誤率1.2%, 可見效果還不錯!
這里把整個kNN.py文件貼出來,主要是上面已經介紹的函數
''' Input: inX: vector to compare to existing dataset (1xN) dataSet: size m data set of known vectors (NxM) labels: data set labels (1xM vector) k: number of neighbors to use for comparison (should be an odd number) Output: the most popular class label ''' 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] def createDataSet(): group = array([[1.0,1.1],[1.0,1.0],[0,0],[0,0.1]]) labels = ['A','A','B','B'] return group, labels def file2matrix(filename): fr = open(filename) numberOfLines = len(fr.readlines()) #get the number of lines in the file returnMat = zeros((numberOfLines,3)) #prepare matrix to return classLabelVector = [] #prepare labels return fr = open(filename) index = 0 for line in fr.readlines(): line = line.strip() listFromLine = line.split('\t') returnMat[index,:] = listFromLine[0:3] classLabelVector.append(int(listFromLine[-1])) index += 1 return returnMat,classLabelVector def autoNorm(dataSet): minVals = dataSet.min(0) maxVals = dataSet.max(0) ranges = maxVals - minVals normDataSet = zeros(shape(dataSet)) m = dataSet.shape[0] normDataSet = dataSet - tile(minVals, (m,1)) normDataSet = normDataSet/tile(ranges, (m,1)) #element wise divide return normDataSet, ranges, minVals def datingClassTest(): hoRatio = 0.50 #hold out 10% datingDataMat,datingLabels = file2matrix('datingTestSet2.txt') #load data setfrom file normMat, ranges, minVals = autoNorm(datingDataMat) m = normMat.shape[0] numTestVecs = int(m*hoRatio) errorCount = 0.0 for i in range(numTestVecs): classifierResult = classify0(normMat[i,:],normMat[numTestVecs:m,:],datingLabels[numTestVecs:m],3) print "the classifier came back with: %d, the real answer is: %d" % (classifierResult, datingLabels[i]) if (classifierResult != datingLabels[i]): errorCount += 1.0 print "the total error rate is: %f" % (errorCount/float(numTestVecs)) print errorCount def classifyPerson(): resultList = ['not at all', 'in small doses', 'in large doses'] percentTats = float(raw_input("percentage of time spent playing video games?")) ffMiles = float(raw_input("frequent flier miles earned per year?")) iceCream = float(raw_input("liters of ice cream consumed per year?")) datingDataMat, datingLabels = file2matrix('datingTestSet2.txt') normMat, ranges, minVals = autoNorm(datingDataMat) inArr = array([ffMiles, percentTats, iceCream]) classifierResult = classify0((inArr-minVals)/ranges, normMat, datingLabels,3) print "You will probably like this person" , resultList[classifierResult - 1] def img2vector(filename): returnVect = zeros((1,1024)) fr = open(filename) for i in range(32): lineStr = fr.readline() for j in range(32): returnVect[0,32*i+j] = int(lineStr[j]) return returnVect def handwritingClassTest(): hwLabels = [] trainingFileList = listdir('trainingDigits') #load the training set m = len(trainingFileList) trainingMat = zeros((m,1024)) for i in range(m): fileNameStr = trainingFileList[i] fileStr = fileNameStr.split('.')[0] #take off .txt classNumStr = int(fileStr.split('_')[0]) hwLabels.append(classNumStr) trainingMat[i,:] = img2vector('trainingDigits/%s' % fileNameStr) testFileList = listdir('testDigits') #iterate through the test set errorCount = 0.0 mTest = len(testFileList) for i in range(mTest): fileNameStr = testFileList[i] fileStr = fileNameStr.split('.')[0] #take off .txt classNumStr = int(fileStr.split('_')[0]) vectorUnderTest = img2vector('testDigits/%s' % fileNameStr) classifierResult = classify0(vectorUnderTest, trainingMat, hwLabels, 3) print "the classifier came back with: %d, the real answer is: %d" % (classifierResult, classNumStr) if (classifierResult != classNumStr): errorCount += 1.0 print "\nthe total number of errors is: %d" % errorCount print "\nthe total error rate is: %f" % (errorCount/float(mTest))
結尾
至此,這個k-近鄰算法的介紹到這里就結束了,希望這篇文章對你的學習有幫助!
百度雲鏈接: https://pan.baidu.com/s/1OuyOuGi9r8eaPS9gglAzBg