樸素貝葉斯Python實例及解析
更新時間:2018年11月19日 16:32:02 作者:qq_36047182
這篇文章主要為大家詳細介紹了樸素貝葉斯Python算法實現(xiàn),具有一定的參考價值,感興趣的小伙伴們可以參考一下
本文實例為大家分享了Python樸素貝葉斯實例代碼,供大家參考,具體內(nèi)容如下
#-*- coding: utf-8 -*- #添加中文注釋 from numpy import * #過濾網(wǎng)站的惡意留言 #樣本數(shù)據(jù) 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']] #類別標(biāo)簽:1侮辱性文字,0正常言論 classVec = [0,1,0,1,0,1] #返回文檔向量,類別向量 return postingList,classVec #創(chuàng)建詞匯表 #輸入:dataSet已經(jīng)經(jīng)過切分處理 #輸出:包含所有文檔中出現(xiàn)的不重復(fù)詞的列表 def createVocabList(dataSet): #構(gòu)建set集合,會返回不重復(fù)詞表 vocabSet = set([]) #遍歷每篇文檔向量,掃描所有文檔的單詞 for document in dataSet: #通過set(document),獲取document中不重復(fù)詞列表 vocabSet = vocabSet | set(document) #求并集 return list(vocabSet) #***詞集模型:只考慮單詞是否出現(xiàn) #vocabList:詞匯表 #inputSet :某個文檔向量 def setOfWords2Vec(vocabList, inputSet): #創(chuàng)建所含元素全為0的向量 returnVec = [0]*len(vocabList) #依次取出文檔中的單詞與詞匯表進行對照,若在詞匯表中出現(xiàn)則為1 for word in inputSet: if word in vocabList: #單詞在詞匯表中出現(xiàn),則記為1 returnVec[vocabList.index(word)] = 1 #詞集模型 #若測試文檔的單詞,不在詞匯表中,顯示提示信息,該單詞出現(xiàn)次數(shù)用0表示 else: print "the word: %s is not in my Vocabulary!" % word return returnVec #====訓(xùn)練分類器,原始的樸素貝葉斯,沒有優(yōu)化===== #輸入trainMatrix:詞向量數(shù)據(jù)集 #輸入trainCategory:數(shù)據(jù)集對應(yīng)的類別標(biāo)簽 #輸出p0Vect:詞匯表中各個單詞在正常言論中的類條件概率密度 #輸出p1Vect:詞匯表中各個單詞在侮辱性言論中的類條件概率密度 #輸出pAbusive:侮辱性言論在整個數(shù)據(jù)集中的比例 def trainNB00(trainMatrix,trainCategory): #numTrainDocs訓(xùn)練集總條數(shù) numTrainDocs = len(trainMatrix) #訓(xùn)練集中所有不重復(fù)單詞總數(shù) numWords = len(trainMatrix[0]) #侮辱類的概率(侮辱類占總訓(xùn)練數(shù)據(jù)的比例) pAbusive = sum(trainCategory)/float(numTrainDocs) #*正常言論的類條件概率密度 p(某單詞|正常言論)=p0Num/p0Denom p0Num = zeros(numWords); #初始化分子為0 #*侮辱性言論的類條件概率密度 p(某單詞|侮辱性言論)=p1Num/p1Denom p1Num = zeros(numWords) #初始化分子為0 #初始化分母置為0 p0Denom = 0; p1Denom = 0 #遍歷訓(xùn)練集數(shù)據(jù) for i in range(numTrainDocs): #若為侮辱類 if trainCategory[i] == 1: #統(tǒng)計侮辱類所有文檔中的各個單詞總數(shù) p1Num += trainMatrix[i] #p1Denom侮辱類總單詞數(shù) p1Denom += sum(trainMatrix[i]) #若為正常類 else: #統(tǒng)計正常類所有文檔中的各個單詞總數(shù) p0Num += trainMatrix[i] #p0Denom正常類總單詞數(shù) p0Denom += sum(trainMatrix[i]) #詞匯表中的單詞在侮辱性言論文檔中的類條件概率 p1Vect = p1Num/p1Denom #詞匯表中的單詞在正常性言論文檔中的類條件概率 p0Vect = p0Num/p0Denom return p0Vect,p1Vect,pAbusive #=====訓(xùn)練分類器,優(yōu)化處理===== #輸入trainMatrix:詞向量數(shù)據(jù)集 #輸入trainCategory:數(shù)據(jù)集對應(yīng)的類別標(biāo)簽 #輸出p0Vect:詞匯表中各個單詞在正常言論中的類條件概率密度 #輸出p1Vect:詞匯表中各個單詞在侮辱性言論中的類條件概率密度 #輸出pAbusive:侮辱性言論在整個數(shù)據(jù)集中的比例 def trainNB0(trainMatrix,trainCategory): #訓(xùn)練集總條數(shù):行數(shù) numTrainDocs = len(trainMatrix) #訓(xùn)練集中所有單詞總數(shù):詞向量維度 numWords = len(trainMatrix[0]) #侮辱類的概率(侮辱類占總訓(xùn)練數(shù)據(jù)的比例) pAbusive = sum(trainCategory)/float(numTrainDocs) #*拉普拉斯平滑防止類條件概率為0,初始化分子為1,分母為2 #正常類向量置為1 p0Num = ones(numWords); #初始化分子為1 #侮辱類向量置為1 p1Num = ones(numWords) #初始化分子為1 #初始化分母置為2 p0Denom = 2.0; p1Denom = 2.0 #遍歷訓(xùn)練集每個樣本 for i in range(numTrainDocs): #若為侮辱類 if trainCategory[i] == 1: #統(tǒng)計侮辱類所有文檔中的各個單詞總數(shù) p1Num += trainMatrix[i] #向量 #p1Denom侮辱類總單詞數(shù) p1Denom += sum(trainMatrix[i]) #若為正常類 else: #統(tǒng)計正常類所有文檔中的各個單詞總數(shù) p0Num += trainMatrix[i] #p0Denom正常類總單詞數(shù) p0Denom += sum(trainMatrix[i]) #數(shù)據(jù)取log,即單個單詞的p(x1|c1)取log,防止下溢出 p1Vect = log(p1Num/p1Denom) p0Vect = log(p0Num/p0Denom) return p0Vect,p1Vect,pAbusive #vec2Classify:待分類文檔 #p0Vect:詞匯表中每個單詞在訓(xùn)練樣本的正常言論中的類條件概率密度 #p1Vect:詞匯表中每個單詞在訓(xùn)練樣本的侮辱性言論中的類條件概率密度 #pClass1:侮辱性言論在訓(xùn)練集中所占的比例 def classifyNB(vec2Classify, p0Vect, p1Vect, pClass1): #在對數(shù)空間中進行計算,屬于哪一類的概率比較大就判為哪一類 #print'0p1=',sum(vec2Classify * p0Vect) #查看結(jié)果 #print'0p0=',sum(vec2Classify * p0Vect) p1 = sum(vec2Classify * p1Vect) + log(pClass1) p0 = sum(vec2Classify * p0Vect) + log(1.0 - pClass1) #print'p1=',p1 #print'p0=',p0 if p1 > p0: return 1 else: return 0 def testingNB(): #獲得訓(xùn)練數(shù)據(jù),類別標(biāo)簽 listOPosts,listClasses = loadDataSet() #創(chuàng)建詞匯表 myVocabList = createVocabList(listOPosts) #構(gòu)建矩陣,存放訓(xùn)練數(shù)據(jù) trainMat=[] #遍歷原始數(shù)據(jù),轉(zhuǎn)換為詞向量,構(gòu)成數(shù)據(jù)訓(xùn)練矩陣 for postinDoc in listOPosts: #數(shù)據(jù)轉(zhuǎn)換后存入數(shù)據(jù)訓(xùn)練矩陣trainMat中 trainMat.append(setOfWords2Vec(myVocabList, postinDoc)) #訓(xùn)練分類器 p0V,p1V,pAb = trainNB0(array(trainMat),array(listClasses)) #===測試數(shù)據(jù)(1) testEntry = ['love', 'my', 'dalmation'] #測試數(shù)據(jù)轉(zhuǎn)為詞向量 thisDoc = array(setOfWords2Vec(myVocabList, testEntry)) #輸出分類結(jié)果 print testEntry,'classified as: ',classifyNB(thisDoc,p0V,p1V,pAb) #===測試數(shù)據(jù)(2) testEntry = ['stupid', 'garbage'] #測試數(shù)據(jù)轉(zhuǎn)為詞向量 thisDoc = array(setOfWords2Vec(myVocabList, testEntry)) #輸出分類結(jié)果 print testEntry,'classified as: ',classifyNB(thisDoc,p0V,p1V,pAb) #***詞袋模型:考慮單詞出現(xiàn)的次數(shù) #vocabList:詞匯表 #inputSet :某個文檔向量 def bagOfWords2VecMN(vocabList, inputSet): #創(chuàng)建所含元素全為0的向量 returnVec = [0]*len(vocabList) #依次取出文檔中的單詞與詞匯表進行對照,統(tǒng)計單詞在文檔中出現(xiàn)的次數(shù) for word in inputSet: if word in vocabList: #單詞在文檔中出現(xiàn)的次數(shù) returnVec[vocabList.index(word)] += 1 #若測試文檔的單詞,不在詞匯表中,顯示提示信息,該單詞出現(xiàn)次數(shù)用0表示 else: print "the word: %s is not in my Vocabulary!" % word return returnVec #準(zhǔn)備數(shù)據(jù),按空格切分出詞 #單詞長度小于或等于2的全部丟棄 def textParse(bigString): import re listOfTokens = re.split(r'\W*', bigString) #tok.lower() 將整個詞轉(zhuǎn)換為小寫 return [tok.lower() for tok in listOfTokens if len(tok) > 2] def spamTest(): #文章按篇存放 docList=[]; #存放文章類別 classList = []; #存放所有文章內(nèi)容 fullText =[] for i in range(1,26): #讀取垃圾郵件 #wordList = textParse(open('D:/work/python/email/spam/%d.txt' % i).read()) wordList = textParse(open('D:/machine learning/python/bayes/email/spam/%d.txt' % i).read()) #docList按篇存放文章 docList.append(wordList) #fullText郵件內(nèi)容存放到一起 fullText.extend(wordList) #垃圾郵件類別標(biāo)記為1 classList.append(1) #讀取正常郵件 #wordList = textParse(open('D:/work/python/email/ham/%d.txt' % i).read()) wordList = textParse(open('D:/machine learning/python/bayes/email/ham/%d.txt' % i).read()) docList.append(wordList) fullText.extend(wordList) #正常郵件類別標(biāo)記為0 classList.append(0) #創(chuàng)建詞典 vocabList = createVocabList(docList) #訓(xùn)練集共50篇文章 trainingSet = range(50); #創(chuàng)建測試集 testSet=[] #隨機選取10篇文章為測試集,測試集中文章從訓(xùn)練集中刪除 for i in range(10): #0-50間產(chǎn)生一個隨機數(shù) randIndex = int(random.uniform(0,len(trainingSet))) #從訓(xùn)練集中找到對應(yīng)文章,加入測試集中 testSet.append(trainingSet[randIndex]) #刪除對應(yīng)文章 del(trainingSet[randIndex]) #準(zhǔn)備數(shù)據(jù),用于訓(xùn)練分類器 trainMat=[]; #訓(xùn)練數(shù)據(jù) trainClasses = [] #類別標(biāo)簽 #遍歷訓(xùn)練集中文章數(shù)據(jù) for docIndex in trainingSet: #每篇文章轉(zhuǎn)為詞袋向量模型,存入trainMat數(shù)據(jù)矩陣中 trainMat.append(bagOfWords2VecMN(vocabList, docList[docIndex])) #trainClasses存放每篇文章的類別 trainClasses.append(classList[docIndex]) #訓(xùn)練分類器 p0V,p1V,pSpam = trainNB0(array(trainMat),array(trainClasses)) #errorCount記錄測試數(shù)據(jù)出錯次數(shù) errorCount = 0 #遍歷測試數(shù)據(jù)集,每條數(shù)據(jù)相當(dāng)于一條文本 for docIndex in testSet: #文本轉(zhuǎn)換為詞向量模型 wordVector = bagOfWords2VecMN(vocabList, docList[docIndex]) #模型給出的分類結(jié)果與本身類別不一致時,說明模型出錯,errorCount數(shù)加1 if classifyNB(array(wordVector),p0V,p1V,pSpam) != classList[docIndex]: errorCount += 1 #輸出出錯的文章 print "classification error",docList[docIndex] #輸出錯誤率,即出錯次數(shù)/總測試次數(shù) print 'the error rate is: ',float(errorCount)/len(testSet) #return vocabList,fullText if __name__ == "__main__": ###**********************留言板數(shù)據(jù):觀察參數(shù)值start ### #獲取數(shù)據(jù) listOPosts,listClasses = loadDataSet() # #構(gòu)建詞匯表 myVocabList = createVocabList(listOPosts) print 'myVocabList=',myVocabList print 'result=',setOfWords2Vec(myVocabList, listOPosts[0]) trainMat = [] for postinDoc in listOPosts: #構(gòu)建訓(xùn)練矩陣 trainMat.append(setOfWords2Vec(myVocabList, postinDoc)) p0Vect,p1Vect,pAbusive = trainNB0(trainMat, listClasses) print 'p0Vect=' print p0Vect print 'p1Vect=' print p1Vect print 'pAbusive=' print pAbusive print 'trainMatrix=' print trainMat print 'listClasses=',listClasses ###**********************留言板數(shù)據(jù):觀察參數(shù)值end ## #測試留言板文檔 print'===================================' testingNB() #***********************垃圾郵件 ## #垃圾郵件分類 print'=======spam filtering=============' spamTest()
以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
您可能感興趣的文章:
相關(guān)文章
python可視化篇之流式數(shù)據(jù)監(jiān)控的實現(xiàn)
這篇文章主要介紹了python可視化篇之流式數(shù)據(jù)監(jiān)控的實現(xiàn),文中通過示例代碼介紹的非常詳細,對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-08-08Django框架組成結(jié)構(gòu)、基本概念與文件功能分析
這篇文章主要介紹了Django框架組成結(jié)構(gòu)、基本概念與文件功能,簡單分析了Django框架的基本架構(gòu)、組成、項目創(chuàng)建方法、模塊組成以及相關(guān)文件功能,需要的朋友可以參考下2019-07-07關(guān)于Python核心框架tornado的異步協(xié)程的2種方法詳解
今天小編就為大家分享一篇關(guān)于Python核心框架tornado的異步協(xié)程的2種方法詳解,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08PyTorch加載預(yù)訓(xùn)練模型實例(pretrained)
今天小編就為大家分享一篇PyTorch加載預(yù)訓(xùn)練模型實例(pretrained),具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-01-01用Python實現(xiàn)web端用戶登錄和注冊功能的教程
這篇文章主要介紹了用Python實現(xiàn)web端用戶登錄和注冊功能的教程,需要的朋友可以參考下2015-04-04