朴素贝叶斯分类java实现 朴素贝叶斯分类代码

机器学习实战（Machine Learning in Action）学习笔记————04.朴素贝叶斯分类（bayes）

关键字：朴素贝叶斯、python、源码解析
作者：米仓山下
时间：2018-10-25
机器学习实战（Machine Learning in Action,@author: Peter Harrington）
源码下载地址：https://www.manning.com/books/machine-learning-in-action
[e-mail]:pbharrin/machinelearninginaction.git

*************************************************************
一、朴树贝叶斯分类（bayes）

#朴素贝叶斯实现文本分类原理：
bayes公式：P(ci|w)=P(w|ci)P(ci)/p(w)
首先计算侮辱性言论和非侮辱性言论文档出现的概率P(ci)即P(1)、P(0)的概率；接着计算P(w|ci)，假设词的出现相互独立（也是朴素贝叶斯“朴素”的含义），p(w0,w1,…wN|c1)=p(w0|c1)p(w1|c1)…p(wN|c1)，然后利用公式计算在该词向量下属于不同类别概率，比较返回概率最大的类别

训练：P(w|ci)——分类ci下词向量w的条件概率，P(w|ci)=p(w0,w1,…wN|c1)=p(w0|c1)p(w1|c1)…p(wN|c1)；P(ci)——分类ci出现的概率；p(w)——词向量w出现的概率（每个词出现的概率之积，是一个定值？）求解：P(ci|w)——词向量w下分类为ci的条件概率

测试：利用公式计算在该词向量下属于不同类别概率，比较返回概率最大的类别

--------------------------------------------------------------
#朴素贝叶斯实现文本分类，训练函数
#input：trainMatrix——训练样本（词集模型，词出现与否分别为1和0），trainCategory——类别向量
#output：p0Vect——[2],p1Vect——[1],pAbusive——样本为侮辱性P(1)的概率

def trainNB0(trainMatrix,trainCategory):
    numTrainDocs = len(trainMatrix)                     #训练样本数目
    numWords = len(trainMatrix[0])                      #特征个数（词汇表大小）
    pAbusive = sum(trainCategory)/float(numTrainDocs)   #样本为侮辱性P(1)的概率，即P(ci)
    p0Num = ones(numWords); p1Num = ones(numWords)      #初始化概率，[注1]
    p0Denom = 2.0; p1Denom = 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)           #[1]侮辱性样本中，词汇表每个词汇出现概率，即P(w|ci)，[注1]
    p0Vect = log(p0Num/p0Denom)           #[2]非侮辱性样本中，词汇表每个词汇出现概率，同为P(w|ci)
    return p0Vect,p1Vect,pAbusive

#注：利用贝叶斯分类器对文档进行分类时，要计算多个概率的乘积以获取文档属于某个类别的概率（p(w0|c=1)p(w1|c=1)p(w2|c=1)）,若其中一个概率为零则整体为零，为避免影响，将p0Num，p1Num初始化为1，将分母p0Denom初始化为2；另外为避免太多小数相乘造成数据下溢，因此对其取对数（f(x)与ln(f(x))具有相同的单调性），并有ln(a*b)=ln(a)+n(b),将其后的乘积运算转换为加和

#朴素贝叶斯分类测试函数

def classifyNB(vec2Classify, p0Vec, p1Vec, pClass1):    #计算属于不同类别的概率并进行比较
    p1 = sum(vec2Classify * p1Vec) + log(pClass1)       #计算概率p(w0|c1)p(w1|c1)…p(wN|c1)p(c1)，
    p0 = sum(vec2Classify * p0Vec) + log(1.0 - pClass1) #因为进行了对数运算，乘积运算转换为加和
    if p1 > p0:
        return 1
    else:
        return 0

-------------------------------------------------------------
测试：

>>> import bayes
>>> from numpy import *
>>> listOPosts,listClasses=bayes.loadDataSet()     #读取训练数据集及类别标签
>>> listOPosts
[['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']]
>>> listClasses
[0, 1, 0, 1, 0, 1]
>>> myVocalList=bayes.createVocabList(listOPosts)  #创建词汇表
>>> myVocalList
['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']
>>> trainMat=[]
>>> for postindoc in listOPosts:                     #将训练数据集转换为词向量形式
...   trainMat.append(bayes.setOfWords2Vec(myVocalList,postindoc))
...
>>> trainMat
[[0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 0, 1], [0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 1, 1, 0, 0, 0, 0, 0, 1, 1, 0, 1, 0, 1, 0, 1, 0, 0, 0], [1, 1, 0, 0, 0, 1, 0, 1, 0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 1], [0, 0, 0, 1, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 1, 0, 1, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 1, 0, 0, 1, 0, 0, 0, 0, 1, 1, 0, 0, 0, 1, 0, 0, 0, 1, 1, 1], [0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 1, 0, 0, 1, 0, 0, 0, 0, 1, 0, 1, 0, 0, 0, 0, 0]]
>>> p0v,p1v,pab=bayes.trainNB0(trainMat,listClasses)#计算条件概率及侮辱性文档出现的概率p(1)
>>> p0v,p1v,pab
(array([-2.56494936, -2.56494936, -2.56494936, -3.25809654, -3.25809654,
       -2.56494936, -2.56494936, -2.56494936, -3.25809654, -2.56494936,
       -2.56494936, -2.56494936, -2.56494936, -3.25809654, -3.25809654,
       -2.15948425, -3.25809654, -3.25809654, -2.56494936, -3.25809654,
       -2.56494936, -2.56494936, -3.25809654, -2.56494936, -2.56494936,
       -2.56494936, -3.25809654, -2.56494936, -3.25809654, -2.56494936,
       -2.56494936, -1.87180218]), array([-3.04452244, -3.04452244, -3.04452244, -2.35137526, -2.35137526,
       -3.04452244, -3.04452244, -3.04452244, -2.35137526, -2.35137526,
       -3.04452244, -3.04452244, -3.04452244, -2.35137526, -2.35137526,
       -2.35137526, -2.35137526, -2.35137526, -3.04452244, -1.94591015,
       -3.04452244, -2.35137526, -2.35137526, -3.04452244, -1.94591015,
       -3.04452244, -1.65822808, -3.04452244, -2.35137526, -3.04452244,
       -3.04452244, -3.04452244]), 0.5)
>>>
>>> testEntry = ['love', 'my', 'dalmation']
>>> thisDoc = array(bayes.setOfWords2Vec(myVocalList, testEntry))  #测试数据转词向量
>>> thisDoc
array([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])
>>> print testEntry,'classified as: ',bayes.classifyNB(thisDoc,p0v,p1v,pab)#判断测试数据类别
['love', 'my', 'dalmation'] classified as:  0

*************************************************************
二、示例：朴树贝叶斯过滤垃圾邮件
email文件夹下有两个文件夹ham和spam，分别存放垃圾邮件和非垃圾邮件，将这50条数据，40条作为训练样本，10条数据作为测试样本，使用词袋模型完成训练和测试，并打印错误词，计算错误率

>>> import bayes
>>> bayes.spamTest()
classification error ['benoit', 'mandelbrot', '1924', '2010', 'benoit', 'mandelbrot', '1924', '2010', 'wilmott', 'team', 'benoit', 'mandelbrot', 'the', 'mathematician', 'the', 'father', 'fractal', 'mathematics', 'and', 'advocate', 'more', 'sophisticated', 'modelling', 'quantitative', 'finance', 'died', '14th', 'october', '2010', 'aged', 'wilmott', 'magazine', 'has', 'often', 'featured', 'mandelbrot', 'his', 'ideas', 'and', 'the', 'work', 'others', 'inspired', 'his', 'fundamental', 'insights', 'you', 'must', 'logged', 'view', 'these', 'articles', 'from', 'past', 'issues', 'wilmott', 'magazine']
classification error ['yeah', 'ready', 'may', 'not', 'here', 'because', 'jar', 'jar', 'has', 'plane', 'tickets', 'germany', 'for']
classification error ['home', 'based', 'business', 'opportunity', 'knocking', 'your', 'door', 'don', 'rude', 'and', 'let', 'this', 'chance', 'you', 'can', 'earn', 'great', 'income', 'and', 'find', 'your', 'financial', 'life', 'transformed', 'learn', 'more', 'here', 'your', 'success', 'work', 'from', 'home', 'finder', 'experts']
the error rate is:  0.3
>>>

另一个例子：使用朴素贝叶斯来发现低于相关的用词
原理：利用训练样本求得p0Vect——[2],p1Vect——[1],pAbusive；即P(w|ci)和P(ci)，通过调整移除的高频词数据，达到较高的准确率，然后返回出现概率较大的词汇。
（代码略）
详细见书P70