機器學習實戰1:朴素貝葉斯模型:文本分類+垃圾郵件分類


  學習了那么多機器學習模型,一切都是為了實踐,動手自己寫寫這些模型的實現對自己很有幫助的,堅持,共勉。本文主要致力於總結貝葉斯實戰中程序代碼的實現(python)及朴素貝葉斯模型原理的總結。python的numpy包簡化了很多計算,另外本人推薦使用pandas做數據統計。

一 引言

  讓你猜測一個身高2.16的人的職業,你一般會猜測他是籃球運動員。這個原理就是朴素貝葉斯原理,因為籃球運動員大多身高很高,所以這個人具有籃球運動員的條件,則猜測他是籃球運動員。

  同理,另一個升高1.58的人,你應該不會猜他是籃球運動員。

二 理論

  條件貝葉斯公式:p(Ci | x,y)=p(x,y | Ci)*p(Ci) / p(x,y)

  計算每個類別的概率,若p(C1 | x,y) > p(~C1 | x,y), 則類別屬於類C1,否則不屬於類C1。

  程序中在模型訓練的時候,只需要先在訓練樣本中計算好先驗概率 p(Ci) 條件概率 p(x,y | Ci) 即可,因為p(x,y)不隨Ci變化,不影響p(Ci | x,y)的最好大小。

  :條件貝葉斯是保證條件之間獨立的(文檔分類中是假設一個詞匯出現與其他詞匯是否出現無關,然而同一主題的詞匯一起出現的概率很高,存在關聯),所以這個假設過於簡單;盡管如此,然而事實表明,朴素貝葉斯的效果還很好。

三 實戰1 -文本分類(應用過濾惡意留言等)

  下面是二分類問題,文檔只能屬於0和1兩個類別,

  1 載入數據集:6條文本及它們各自的類別,這6條文本作為訓練集。

from numpy import *

def loadDataSet():
    postingList=[['my', 'dog', 'has', 'flea', 'problems', 'help', 'please'],
                 ['maybe', 'not', 'take', 'him', 'to', 'dog', 'park', 'stupid'],
                 ['my', 'dalmation', 'is', 'so', 'cute', 'I', 'love', 'him'],
                 ['stop', 'posting', 'stupid', 'worthless', 'garbage'],
                 ['mr', 'licks', 'ate', 'my', 'steak', 'how', 'to', 'stop', 'him'],
                 ['quit', 'buying', 'worthless', 'dog', 'food', 'stupid']]
    classVec = [0,1,0,1,0,1]    #1 is abusive, 0 not
    return postingList,classVec

  2 創建詞匯表:利用集合結構內元素的唯一性,創建一個包含所有詞匯的詞表。

def createVocabList(dataSet):
    vocabSet = set([])  #create empty set
    for document in dataSet:
        vocabSet = vocabSet | set(document) #union of the two sets
    return list(vocabSet)

  3 把輸入文本根據詞表轉化為計算機可處理的01向量形式:

  eq,測試文本1: ['love', 'my', 'dalmation']

    詞匯表:['cute', 'love', 'help', 'garbage', 'quit', 'I', 'problems', 'is', 'park', 'stop', 'flea', 'dalmation', 'licks', 'food', 'not', 'him', 'buying', 'posting', 'has', 'worthless', 'ate', 'to', 'maybe', 'please', 'dog', 'how', 'stupid', 'so', 'take', 'mr', 'steak', 'my']

    向量化結果:[0 1 0 0 0 0 0 0 0 0 0 1 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 1]

def setOfWords2Vec(vocabList, inputSet):
    returnVec = [0]*len(vocabList)
    for word in inputSet:
        if word in vocabList:
            returnVec[vocabList.index(word)] = 1
        else: print "the word: %s is not in my Vocabulary!" % word
    return returnVec

  

  4訓練模型:在訓練樣本中計算先驗概率 p(Ci)條件概率 p(x,y | Ci),本實例有0和1兩個類別,所以返回p(x,y | 0),p(x,y | 1)和p(Ci)。

  此處有兩個改進的地方:

    (1)若有的類別沒有出現,其概率就是0,會十分影響分類器的性能。所以采取各類別默認1次累加,總類別(兩類)次數2,這樣不影響相對大小。

    (2)若很小是數字相乘,則結果會更小,再四舍五入存在誤差,而且會造成下溢出。采取取log,乘法變為加法,並且相對大小趨勢不變。

def trainNB0(trainMatrix,trainCategory):
    numTrainDocs = len(trainMatrix)
    numWords = len(trainMatrix[0])
    pAbusive = sum(trainCategory)/float(numTrainDocs)
    p0Num = ones(numWords); p1Num = ones(numWords)      #change to ones() 
    p0Denom = 2.0; p1Denom = 2.0                        #change to 2.0
    for i in range(numTrainDocs):
        if trainCategory[i] == 1:        
            p1Num += trainMatrix[i]
            p1Denom += sum(trainMatrix[i])
        else:
            p0Num += trainMatrix[i]
            p0Denom += sum(trainMatrix[i])
    p1Vect = log(p1Num/p1Denom)          #change to log()
    p0Vect = log(p0Num/p0Denom)          #change to log()

    return p0Vect,p1Vect,pAbusive

  5 分類:根據計算后,哪個類別的概率大,則屬於哪個類別。

def classifyNB(vec2Classify, p0Vec, p1Vec, pClass1):
    p1 = sum(vec2Classify * p1Vec) + log(pClass1)    #element-wise mult
    p0 = sum(vec2Classify * p0Vec) + log(1.0 - pClass1)
    if p1 > p0:
        return 1
    else: 
        return 0

  6 測試函數:

    加載數據集+提煉詞表;

    訓練模型:根據六條訓練集計算先驗概率和條件概率;

    測試模型:對訓練兩條測試文本進行分類。

def testingNB():
    listOPosts,listClasses = loadDataSet()
    myVocabList = createVocabList(listOPosts)
    trainMat=[]
    for postinDoc in listOPosts:
        trainMat.append(setOfWords2Vec(myVocabList, postinDoc))
    p0V,p1V,pAb = trainNB0(array(trainMat),array(listClasses))
    testEntry = ['love', 'my', 'dalmation']
    thisDoc = array(setOfWords2Vec(myVocabList, testEntry))
    print testEntry,'classified as: ',classifyNB(thisDoc,p0V,p1V,pAb)
    testEntry = ['stupid', 'garbage']
    thisDoc = array(setOfWords2Vec(myVocabList, testEntry))
    print testEntry,'classified as: ',classifyNB(thisDoc,p0V,p1V,pAb)

  缺點:詞表只能記錄詞匯是否出現,不能體現這個詞匯出現的次數。改進方法:采用詞袋模型,見下面垃圾郵件分類實戰。

 

四 實戰2-垃圾郵件分類

  1 對郵件的文本划分成詞匯,長度小於2的默認為不是詞匯,過濾掉即可。返回一串小寫的拆分后的郵件信息。

def textParse(bigString):    #input is big string, #output is word list
    import re
    listOfTokens = re.split(r'\W*', bigString)
    return [tok.lower() for tok in listOfTokens if len(tok) > 2] 

  2 文檔詞袋模型:使用數組代替集合數據結構,可以保存詞匯頻率信息。

def bagOfWords2VecMN(vocabList, inputSet):
    returnVec = [0]*len(vocabList)
    for word in inputSet:
        if word in vocabList:
            returnVec[vocabList.index(word)] += 1
    return returnVec

  3 輸入為25封正常郵件和25封垃圾郵件。50封郵件中隨機選取10封作為測試樣本,剩余40封作為訓練樣本。

   訓練模型:40封訓練樣本,訓練出先驗概率條件概率

   測試模型:遍歷10個測試樣本,計算垃圾郵件分類的正確率。

def spamTest():
    docList=[]; classList = []; fullText =[]
    for i in range(1,26):
        wordList = textParse(open('email/spam/%d.txt' % i).read())
        # print wordList
        docList.append(wordList)
        fullText.extend(wordList)
        classList.append(1)
        wordList = textParse(open('email/ham/%d.txt' % i).read())
        docList.append(wordList)
        fullText.extend(wordList)
        classList.append(0)
    vocabList = createVocabList(docList)#create vocabulary
    trainingSet = range(50); testSet=[]           #create test set
    for i in range(10):
        randIndex = int(random.uniform(0,len(trainingSet)))
        testSet.append(trainingSet[randIndex])
        del(trainingSet[randIndex])  
    trainMat=[]; trainClasses = []
    for docIndex in trainingSet:#train the classifier (get probs) trainNB0
        trainMat.append(bagOfWords2VecMN(vocabList, docList[docIndex]))
        trainClasses.append(classList[docIndex])
    p0V,p1V,pSpam = trainNB0(array(trainMat),array(trainClasses))
    errorCount = 0
    for docIndex in testSet:        #classify the remaining items
        wordVector = bagOfWords2VecMN(vocabList, docList[docIndex])
        if classifyNB(array(wordVector),p0V,p1V,pSpam) != classList[docIndex]:
            errorCount += 1
            print "classification error",docList[docIndex]
    print 'the error rate is: ',float(errorCount)/len(testSet)
    #return vocabList,fullText

 

五 小結

  上面我處理的樣本的屬性值都是分類型的,然而數值型的朴素貝葉斯能處理嗎?

  1 朴素貝葉斯處理數值型數據的方法:
  (1) 區間離散化,設閾值,分段。
  (2) 高斯化:求出概率密度函數,假設變量服從正態分布,根據已有變量統計均值和方差,
    得出概率密度函數,這樣就解決了計算連續值作為分類的條件概率值。
      參考:http://blog.mythsman.com/?p=2683

  2 除0問題:
    Laplace校准 所有計算均加一,總類別數目加n;

  3 下溢出:很小的值相乘,四舍五入誤差
    采用log 乘法變相加;

 

  4移除停用詞:也可以提高文本分類的性能

 

 

參考:書<機器學習實戰> Peter

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM