1. 程式人生 > >【Machine :Learning】 樸素貝葉斯

【Machine :Learning】 樸素貝葉斯

1. 樸素貝葉斯: 條件概率在機器學習演算法的應用。理解這個演算法需要一點推導。不會編輯公式。。

核心就是 在已知訓練集的前提條件下,算出每個特徵的概率為該分類的概率, 然後套貝葉斯公式計算 預測集的所有分類概率,預測型別為概率最大的型別

from numpy import *


def loadDataSet():
    """
    Returns:
        postingList: list, 用於測試的靜態資料
        classVec: list, 標籤,與 postingList 對應, 1 代表侮辱性文字, 0 代表正常言論
    
""" 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 代表侮辱性文字, 0 代表正常言論 return postingList, classVec def createVocabList(dataSet): """ 對資料進行去重 Args: dataSet: list, 原始資料 Returns: list 去重後的一維list
""" vocabSet = set([]) for document in dataSet: vocabSet = vocabSet | set(document) return list(vocabSet) def setOfWords2Vec(vocabList, inputSet): """ 對資料使用情況進行標記 Args: vocabList: list 參考資料 inputSet: 測試資料 Returns: list 對應 vocabList, 1 代表在 inputSet 中存在, 0 代表不存在 """ returnVec = [0] * len(vocabList) for word in inputSet: if word in inputSet: returnVec[vocabList.index(word)] = 1 else: print("the word: %s is not my Vocabulary!" % word) return returnVec def trainNB0(trainMatrix, trainCategory): """ Args: trainMatrix: 測試資料 trainCategory: 資料標籤 Returns: p0Vect: list[list] 在已知正常文件的概率是 0.4的前提下, 每個單詞的為正常單詞的概率) p1Vect: list[list] (在已知侮辱性文件的概率是 0.6的前提下, 每個單詞的為侮辱性單詞的概率) pAbusive: float 條件概率中的條件 以 createDataSet 方法中的資料為例, 侮辱性文件的概率是 0.6, 正常文件的概率是0.4 """ numTrainDocs = len(trainMatrix) numWords = len(trainMatrix[0]) pAbusive = sum(trainCategory) / float(numTrainDocs) p0Num = zeros(numWords) p1Num = zeros(numWords) p0Denom = 0.0 p1Denom = 0.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 = p1Num / p1Denom p0Vect = p0Num / p0Denom return p0Vect, p1Vect, pAbusive def classifyNB(vec2Classify, p0Vec, p1Vec, pClass1): """ Args: vec2Classify: list[int] 詞彙表使用標記 p0Vec: list[list] 單個詞彙為正常詞彙的概率 p1Vec: list[list] 單個詞彙為侮辱性詞彙的概率 pClass1: float 文件為侮辱性文件的概率 Returns: 1: 侮辱性文件 2: 正常文件 """ p1 = sum(vec2Classify * p1Vec) + log(pClass1) # 使用log避免 0乘以任何數為0 的尷尬 p0 = sum(vec2Classify * p0Vec) + log(1.0 - pClass1) if p1 > p0: return 1 else: return 0 def testingNB(): listOPosts, listClasses = loadDataSet() myVocabList = createVocabList(listOPosts) trainMat = [] for postinDoc in listOPosts: trainMat.append(setOfWords2Vec(myVocabList, postinDoc)) p0V, p1V, pAb = trainNB0(trainMat, 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)) if __name__ == '__main__': testingNB()