Loading

3-朴素贝叶斯

📐 基于概率论的分类方法:朴素贝叶斯


贝叶斯分类是一类分类算法的总称,这类算法均以贝叶斯定理为基础,故统称为贝叶斯分类。本章首先介绍贝叶斯分类算法的基础——贝叶斯定理。最后,我们通过实例来讨论贝叶斯分类的中最简单的一种: 朴素贝叶斯分类。

1. 基于贝叶斯决策理论的分类方法

朴素贝叶斯是贝叶斯决策理论的一部分:

我们现在有一个数据集,它由两类数据组成,数据分布如下图所示:

我们现在用 p1(x,y) 表示数据点 (x,y) 属于类别 1(图中用圆点表示的类别)的概率,用 p2(x,y) 表示数据点 (x,y) 属于类别 2(图中三角形表示的类别)的概率。

那么对于一个新数据点 (x,y),可以用下面的规则来判断它的类别:

  • 如果 p1(x,y) > p2(x,y) ,那么类别为1
  • 如果 p2(x,y) > p1(x,y) ,那么类别为2

也就是说,我们会选择高概率对应的类别。这就是贝叶斯决策理论的核心思想,即选择具有最高概率的决策

2. 条件概率 conditional probability

① 条件概率

有一个装了 7 块石头的罐子,其中 3 块是白色的,4 块是黑色的。如果从罐子中随机取出一块石头,那么是白色石头的可能性是多少?

显然,取出白色石头的概率为 3/7 ,取到黑色石头的概率是 4/7 。我们使用 P(white) 来表示取到白色石头的概率,其概率值可以通过白色石头数目除以总的石头数目来得到。

如果这 7 块石头如下图所示,放在两个桶中,那么上述概率应该如何计算

要计算 P(white) 或者 P(black) ,显然,石头所在桶的信息是会改变结果的,这就是条件概率 conditional probability。假定计算的是从 B 桶取到白色石头的概率,这个概率可以记作 P(white|bucketB) ,我们称之为 “在已知石头出自 B 桶的条件下,取出白色石头的概率”。很容易得到,P(white|bucketA) 值为 2/4 ,P(white|bucketB) 的值为 1/3 。

条件概率的计算公式如下:

P(white|bucketB) = P(white and bucketB) / P(bucketB)

公式解读:

  • P(white|bucketB):在已知石头出自 B 桶的条件下,取出白色石头的概率

  • P(white and bucketB):取出 B 桶中 白色石头的概率 = 1/ 7

  • P(bucketB):取出 B 桶中石头的概率 3 / 7

② 贝叶斯准则

另外一种有效计算条件概率的方法称为贝叶斯准则。贝叶斯准则告诉我们如何交换条件概率中的条件与结果,即如果已知 P(x|c),要求 P(c|x),那么可以使用下面的计算方法:

OK,我们讨论了条件概率,接下来将其应用到分类器中。👇

3. 使用条件概率来分类

第 1 节提到贝叶斯决策理论需要计算两个概率:p1(x, y) 和 p2(x, y)

  • 如果 p1(x, y) > p2(x, y) ,那么类别为 1
  • 如果 p2(x, y) > p1(x, y) ,那么类别为 2

这并不是贝叶斯决策理论的所有内容。使用 p1( ) 和 p2( ) 只是为了尽可能简化描述,而真正需要计算和比较的是 p(c1|x, y) p(c2|x, y) 。这些符号所代表的具体意义是:给定某个由 x、y 表示的数据点,那么该数据点来自类别 c1 的概率是多少?数据点来自类别 c2 的概率又是多少?注意这些概率与概率 p(x, y|c1) 并不一样,不过可以使用贝叶斯准则来交换概率中条件与结果。具体地,应用贝叶斯准则得到:

⭐ 使用上面这些定义,可以定义贝叶斯分类准则为:

  • 如果 P(c1|x, y) > P(c2|x, y), 那么属于类别 c1;
  • 如果 P(c2|x, y) > P(c1|x, y), 那么属于类别 c2。

使用贝叶斯准则,可以通过已知的三个概率值来计算未知的概率值。后面会给出利用贝叶斯准则来计算概率并对数据进行分类的代码。

下一节将介绍一个简单但功能强大的贝叶斯分类器的应用案例。

4. 使用朴素(navie)贝叶斯进行文档分类

机器学习的一个重要应用就是文档的自动分类。

在文档分类中,整个文档(如一封电子邮件)是实例,而电子邮件中的某些元素则构成特征。我们可以观察文档中出现的词,并把每个词作为一个特征,而每个词的出现或者不出现作为该特征的值,这样得到的特征数目就会跟词汇表中的词的数目一样多。

朴素贝叶斯是上面介绍的贝叶斯分类器的一个扩展,是用于文档分类的常用算法。

朴素贝叶斯的一般过程:

  • 收集数据: 可以使用任何方法。比如 RSS 源
  • 准备数据: 需要数值型或者布尔型数据。
  • 分析数据: 有大量特征时,绘制特征作用不大,此时使用直方图效果更好。
  • 训练算法: 计算不同的独立特征的条件概率。
  • 测试算法: 计算错误率。
  • 使用算法: 一个常见的朴素贝叶斯应用是文档分类。可以在任意的分类场景中使用朴素贝叶斯分类器,不一定非要是文本。

假设词汇表有 1000 个单词 ,要得到好的概率分布,就需要足够的数据样本。如果每个特征需要 N 个样本,那么对于 10 个特征将需要 $N^{10}$ 个样本,对于包含 1000 个特征(单词)的词汇表就需要 $N^{1000}$ 个样本。可以看到,所需要的样本数会随着特征数目的增大而迅速增长。

  • ⚪ 朴素贝叶斯分类器假设特征之间 相互独立 。那么样本数就可以从 $N^{1000}$ 减少到 $1000 × N$。 所谓 独立(independence) 指的是统计意义上的独立,即一个特征或者单词出现的可能性与它和其他单词相邻没有关系,比如说,‘我’ 出现在 ‘们’ 的前面的概率和 出现在 ‘们’ 的后面的概率相同(当然,我们知道这种假设并不正确)。这个假设正是朴素贝叶斯分类器中 朴素(naive) 一词的含义。

  • ⚪ 朴素贝叶斯分类器中的另一个假设是,每个特征同等重要

5. 实例 1:使用朴素贝叶斯辨别侮辱性词汇

项目概述:使用朴素贝叶斯构建一个快速过滤器来屏蔽侮辱性文档。如果某篇文档使用了负面或者侮辱性的语言,那么就将该文档标识为侮辱性文档。对此问题建立两个类别: 侮辱类和非侮辱类,使用 1 和 0 分别表示。

① 准备工作:将文本转换成 0-1 序列

目的:我们需要把文档中的每个单词利用 0 和 1 来表示,这样方便我们进行处理。

📑 词表到向量的转换函数:

这里的数据集是我们手动输入的,一个列表代表一篇文档

def loadDataSet():
    """
    创建数据集
    :return: 单词列表postingList, 所属类别 classVec
    """
    postingList = [['my', 'dog', 'has', 'flea', 'problems', 'help', 'please'], # 文档1
                   ['maybe', 'not', 'take', 'him', 'to', 'dog', 'park', 'stupid'], # 文档2
                   ['my', 'dalmation', 'is', 'so', 'cute', 'I', 'love', 'him'], # 文档3
                   ['stop', 'posting', 'stupid', 'worthless', 'garbage'], # 文档4
                   ['mr', 'licks', 'ate', 'my', 'steak', 'how', 'to', 'stop', 'him'], # 文档5
                   ['quit', 'buying', 'worthless', 'dog', 'food', 'stupid']] # 文档6
    classVec = [0, 1, 0, 1, 0, 1]    # 各篇文档中是否带侮辱性文字:1 代表侮辱性文字,0 代表正常言论
    return postingList, classVec

def createVocabList(dataSet):
    """
    获取所有单词去重后的集合
    :param dataSet: 数据集
    :return: 所有单词的集合(即不含重复元素的单词列表)
    """
    vocabSet = set([])  # 创建一个空集
    for document in dataSet:
        vocabSet = vocabSet | set(document)  # 操作符 | 用于求两个集合的并集
    return list(vocabSet)

def setOfWords2Vec(vocabList, inputSet):
    """
    Desc:
       遍历去重后的词汇表 vocabList,将某篇文档 inputSet 中出现的单词位置置 1
            比如说去重后的词汇表 i love you
            某篇文档 i love my love
            那么该篇文档对应的词汇表序列就是 1 1 1;
            如果某篇文档为 i love
    		那么该篇文档对应的词汇表序列是 1 1
    	其实就是将文档转换成 0-1 序列 
    Params:
        vocabList: 去重后的词汇表
        inputSet: 需要进行转换的文档
    Return:
        returnVec: 该篇文档对应的 0-1 序列
    """
    returnVec = [0] * len(vocabList) # 创建一个和去重词汇表等长的向量,并将其初始化。用来表示文档的 0-1 序列
    
    for word in inputSet: # 对文档中的每个单词
        if word in vocabList: # 如果该单词在去重词汇表中出现
            returnVec[vocabList.index(word)] = 1 # 将该单词对应的位置置为 1
        else:
            print("the word: %s is not in my Vocabulary!" %word)
    return returnVec

看看这些函数的效果:

listOPosts,listClasses = loadDataSet()

myVocabList = createVocabList(listOPosts) # 去重后的词汇表

trainMat = [] 
for postinDoc in listOPosts:
    print(postinDoc)
    trainMat.append(setOfWords2Vec(myVocabList,postinDoc))
    print(trainMat)

解释一下:

② 训练算法:从 0-1 序列计算概率

前面介绍了如何将一组单词转换为一组数字,接下来看看如何使用这些数字计算概率。

现在已经知道一个词是否出现在一篇文档中,也知道该文档所属的类别。接下来我们重写贝叶斯准则,将之前的 x, y 替换为 w,表示这是一个向量,即它由多个值组成。在这个例子中,数值个数与词汇表中的词个数相同。

我们使用上述公式,对每个类计算该值,然后比较这两个概率值的大小。

首先可以通过类别 i (侮辱性留言或者非侮辱性留言)中的文档数除以总的文档数来计算概率 $p(c_i)$ 。接下来计算$ p(w | c_i) $,这里就要用到朴素贝叶斯假设。如果将 w 展开为一个个独立特征,那么就可以将上述概率写作 $p(w_0, w_1, w_2...w_n | c_i) $。这里假设所有词都互相独立,该假设也称作条件独立性假设,它意味着可以使用$ p(w_0 | c_i) × p(w_1 | c_i) × p(w_2 | c_i) × ... × p(w_n | c_i) $来计算上述概率,这样就极大地简化了计算的过程。

💡 其实上式中的分母对于所有类别来说都是一样的,对计算结果不会产生影响,我们只需要计算分子即可。

该函数的伪代码如下:

import numpy as np

def trainNB0(trainMatrix, trainCategory):
    """
    训练数据原版
    :param trainMatrix: 文件单词矩阵 [[1,0,1,1,1....],[],[]...]
    :param trainCategory: 各个文件对应的类别[0,1,1,0....],列表长度等于单词矩阵数,其中的1代表对应的文件是侮辱性文件,0代表不是侮辱性文件
    :return:
    """
   
    numTrainDocs = len(trainMatrix)  # 文件数
    numWords = len(trainMatrix[0])  # 单词数
    pAbusive = sum(trainCategory) / float(numTrainDocs) # 侮辱性文件的出现概率,即 trainCategory 中所有 1 的个数(0 1 相加即得 1 的个数)
    # (非)侮辱性单词在每个文件中出现的次数列表
    p0Num = np.zeros(numWords) # [0,0,0,.....] 非侮辱性单词在每个文件中出现的次数列表
    p1Num = np.zeros(numWords) # [0,0,0,.....] 侮辱性单词出在每个文件中出现的次数列表

    # (非)侮辱性单词在(非)侮辱性文档出现的总数
    p0Denom = 0.0 # 0 非侮辱性词汇在所有非侮辱的文档的出现总数
    p1Denom = 0.0 # 1 侮辱性词汇在所有侮辱性的文档的出现总数
    
    for i in range(numTrainDocs):
        # 是否是侮辱性文件
        if trainCategory[i] == 1:
            # 如果是侮辱性文件,对侮辱性文件的向量进行相加,表示在所有侮辱性文件中,去重词汇表中各个词汇出现的次数
            p1Num += trainMatrix[i] # [0,1,1,....] + [0,1,1,....] -> [0,2,2,...]
            # 对向量中的所有元素进行求和,表示在所有侮辱性文件中去重词汇表中所有词汇出现的次数之和
            p1Denom += sum(trainMatrix[i])
        else:
            # 如果是非侮辱性文件,对非侮辱性文件的向量进行相加,表示在所有非侮辱性文件中,去重词汇表中各个词汇出现的次数
            p0Num += trainMatrix[i]
            # 对向量中的所有元素进行求和,表示在所有非侮辱性文件中去重词汇表中所有词汇出现的次数之和
            p0Denom += sum(trainMatrix[i])
            
    
    # 在1类别下,每个单词出现的概率(条件概率)
    p1Vect = p1Num / p1Denom # [1,2,3,5]/90->[1/90,2/90,3/90,5/90]
    
    # 在0类别下,每个单词出现的概率
    p0Vect = p0Num / p0Denom
    return p0Vect, p1Vect, pAbusive

测试一下该函数:

p0V,p1V,pAb = trainNB0(trainMat,listClasses)

  • pAb 是类别 1 即侮辱性文档的出现概率;
  • p0V 表示在已知文档是类别 0 即非侮辱性文档的条件下,去重词汇表中各个单词的出现概率;
  • p1V 表示在已知文档是类别 1 即侮辱性文档的条件下,去重词汇表中各个单词的出现概率。

显然,去重词汇表中第四个位置即 dog 是在侮辱性文档中出现概率最大的单词,这意味着 dog 是最能表征类别 1(侮辱性文档类)的单词。

👇 不过,使用该函数进行分类之前,还需解决函数中的一些缺陷。

③ 测试算法:根据现实情况修改分类器

在利用贝叶斯分类器对文档进行分类时,要计算多个概率的乘积以获得文档属于某个类别的概率,即计算 $p(w_0|1) * p(w_1|1) * p(w_2|1)$。如果其中一个概率值为 0,那么最后的乘积也为 0。为降低这种影响,可以将所有词的出现数初始化为 1,并将分母初始化为 2 (此处其实是采用了拉普拉斯平滑,即在条件概率和先验 概率的计算公式的分子和分母上分别添加一个 λ 和 $S_jλ$,λ = 1,$S_j$ 表示分类的个数,此处只有两个分类,所以 $S_j = 2$。具体详细可见本网站统计学习方法 - 朴素贝叶斯部分

# 构造单词出现次数列表
p0Num = np.ones(numWords) # [1,1,1,.....]
p1Num = np.ones(numWords) # [1,1,1,.....]

# 整个数据集单词出现总数
p0Denom = 2.0 # 0 非侮辱性词汇出现总数
p1Denom = 2.0 # 1 侮辱性词汇出现总数

另一个遇到的问题是下溢出,这是由于太多很小的数相乘造成的。当计算乘积 $p(w_0|c_i) * p(w_1|c_i) * p(w_2|c_i)... p(w_n|c_i)$ 时,由于大部分因子都非常小,所以程序会下溢出或者得到不正确的答案。(用 Python 尝试相乘许多很小的数,最后四舍五入后会得到 0)。一种解决办法是对乘积取自然对数。在代数中有 ln(a * b) = ln(a) + ln(b), 于是通过求对数可以避免下溢出或者浮点数舍入导致的错误。同时,采用自然对数进行处理不会有任何损失。

p1Vect = log(p1Num / p1Denom)
p0Vect = log(p0Num / p0Denom)

下图给出了函数 f(x) 与 ln(f(x)) 的曲线。可以看出,它们在相同区域内同时增加或者减少,并且在相同点上取到极值。它们的取值虽然不同,但不影响最终结果。

训练函数的优化版本如下:

def trainNB0(trainMatrix, trainCategory):
    """
    训练数据优化版本
    :param trainMatrix: 文件单词矩阵
    :param trainCategory: 文件对应的类别
    :return:
    """
    # 总文件数
    numTrainDocs = len(trainMatrix)
    # 总单词数
    numWords = len(trainMatrix[0])
    # 侮辱性文件的出现概率
    pAbusive = sum(trainCategory) / float(numTrainDocs)
    # 构造单词出现次数列表
    # p0Num 正常的统计
    # p1Num 侮辱的统计
    p0Num = np.ones(numWords)#[0,0......]->[1,1,1,1,1.....]
    p1Num = np.ones(numWords)

    # 整个数据集单词出现总数,2.0根据样本/实际调查结果调整分母的值(2主要是避免分母为0,当然值可以调整)
    # p0Denom 正常的统计
    # p1Denom 侮辱的统计
    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)
    p0Vect = log(p0Num / p0Denom)
    return p0Vect, p1Vect, pAbusive

Ok,现在我们可以来构建完整的分类器了:

# 朴素贝叶斯分类函数
def classifyNB(vec2Classify, p0Vec, p1Vec, pClass1):
    """
    :param vec2Classify: 待测数据[0,1,1,1,1...],即该文档中各个单词是否在词汇表中出现
    :param p0Vec: 即词汇表中的每个单词在非侮辱性文档类别0中出现的概率
    :param p1Vec: 即词汇表中的每个单词在侮辱性文档类别1中出现的概率
    :param pClass1: 该篇文档是侮辱性文件的概率
    :return: 类别1 or 0
    """
    # 公式:p(c/w) = p(w/c)*p(c) / p(w)
    # vec2Classify * p1Vec得到的是p(w/ci)
    # 由于贝叶斯p(w/c)=p(w1,w2,w3..../c1)=p(w1)/p(c1) * p(w2)/p(c1) * .....
    # 在上文中,为了解决下溢出问题,我们将 p0Vec 和 p1Vec 都使用了对数来表示
    # 所以这里的每一项p(w1)/p(c1)是一个对数,所以对应公式中的实数乘法就转换成对数加法
    # 最后还需要乘以p(c)  同样将 p(c) 转成对数,将实数乘法转换成对数加法
    p1 = sum(vec2Classify * p1Vec) + np.log(pClass1) # P(w|c1) * P(c1) ,即贝叶斯准则的分子
    p0 = sum(vec2Classify * p0Vec) + np.log(1.0 - pClass1) # P(w|c0) * P(c0) ,即贝叶斯准则的分子
    if p1 > p0:
        return 1
    else:
        return 0
# 测试朴素贝叶斯算法
def testingNB():
    # 1. 加载数据集
    listOPosts, listClasses = loadDataSet() 
    # 2. 创建单词集合
    myVocabList = createVocabList(listOPosts)  
    # 3. 创建每篇文档对应的 0-1 序列
    trainMat = []
    for postinDoc in listOPosts: 
        # 返回 m*len(myVocabList)的矩阵, 记录的都是0,1信息
        trainMat.append(setOfWords2Vec(myVocabList, postinDoc))
    # 4. 训练数据
    p0V, p1V, pAb = trainNB0(np.array(trainMat), np.array(listClasses))
    # 5. 测试数据
    testEntry = ['love', 'my', 'dalmation'] # 文档
    thisDoc = np.array(setOfWords2Vec(myVocabList, testEntry)) # 待测数据,该文档中各个单词是否在词汇表中出现,如果出现,就将词汇表的对应位置置为1
    print(testEntry, 'classified as: ', classifyNB(thisDoc, p0V, p1V, pAb))
    
    testEntry = ['stupid', 'garbage']
    thisDoc = np.array(setOfWords2Vec(myVocabList, testEntry))
    print(testEntry, 'classified as: ', classifyNB(thisDoc, p0V, p1V, pAb))

④ 词袋模型 bag-of-words model

目前为止,我们将每个词的出现与否作为一个特征 ,这可以被描述为词集模型 set-of-words model,每个的单词只能出现一次。

而在词袋模型中 bag-of-words model,每个单词可以出现多次。下面给出基于词袋模型的朴素贝叶斯代码,它于 setOfWords2Vec 基本相同,但是每遇到一个单词时,它会增加词汇表对应位置的值,而不只是将对应的数值设为 1

#  朴素贝叶斯词袋模型
def bagOfWords2VecMN(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

✅ OK,至此为止,分类器已经构建好了,接下来我们利用该分类器来过滤垃圾邮件。

6. 实例 2:使用朴素贝叶斯过滤垃圾邮件

项目概述:接下来我们完成朴素贝叶斯的一个最著名的应用: 电子邮件垃圾过滤。

开发流程

  • 收集数据: 提供文本文件
  • 准备数据: 将文本文件解析成词条向量
  • 分析数据: 检查词条确保解析的正确性
  • 训练算法: 使用我们之前建立的 trainNB0() 函数
  • 测试算法: 使用朴素贝叶斯进行交叉验证
  • 使用算法: 构建一个完整的程序对一组文档进行分类,将错分的文档输出到屏幕上

① 准备数据:切分文本

前一节介绍了如何创建词向量,并基于这些词向量进行朴素贝叶斯分类的过程。前一节中的词向量是预先给定的,下面介绍如何从文本文档中构建词列表

对于一个文本字符串,可以使用 Python 的 string.split() 方法将其切分:

mySent = 'This book is the best book on Python. I love it!'
mySent.split()

可以看到,标点符号也被当成了词的一部分。可以使用正则表达式来切分句子,其中分割符是除单词、数字外的任意字符串:

import re
listOfTokens = re.split(r'\W+', mySent)
listOfTokens

最后,我们希望所有的单词都是同一形式的,即统一大写或小写:

[tok.lower() for tok in listOfTokens if len(tok) > 0]

现在来看数据集中一封完整的电子邮件的实际处理结果:

emailTest = open('email/ham/6.txt').read()
listOfTokens = re.split(r'\W+', emailTest)

② 测试算法:使用朴素贝叶斯进行交叉验证

下面将文本解析器集成到一个完整分类器中:

def textParse(bigString): 
    '''
    Desc:
        接收一个大字符串并将其解析为字符串列表
    Args:
        bigString -- 大字符串
    Returns:
        去掉少于 2 个字符的字符串,并将所有字符串转换为小写,返回字符串列表
    '''
    import re
    listOfTokens = re.split(r'\W+', bigString)
    return [tok.lower() for tok in listOfTokens if len(tok) > 2]

def spamTest():
    '''
    Desc:
        对贝叶斯垃圾邮件分类器进行自动化处理。
    Args:
        none
    Returns:
        对测试集中的每封邮件进行分类,若邮件分类错误,则错误数加 1,最后返回总的错误百分比。
    '''
    docList = [] # 每封邮件的词汇表,一个一个的列表
    classList = [] # 每篇邮件的分类
    fullText = [] # 所有邮件的词汇表(一个列表)
    # 导入并解析文本文件
    for i in range(1, 26):
        # 导入文件夹 spam 下的文件,切分,解析数据,并归类为 1 类别
        wordList = textParse(open('email/spam/%d.txt' % i, encoding="ISO-8859-1").read())
        docList.append(wordList)
        fullText.extend(wordList)
        classList.append(1)
        # 导入文件夹 ham 下的文件切分,解析数据,并归类为 0 类别
        wordList = textParse(open('email/ham/%d.txt' % i, encoding="ISO-8859-1").read())
        docList.append(wordList)
        fullText.extend(wordList)
        classList.append(0)
                                                             
    vocabList = createVocabList(docList) # 创建词汇表(去重)
    trainingSet = range(50)  # 训练集
    testSet = [] # 测试集
    # 随机构建测试集
    for i in range(10):
        # random.uniform(x, y) 随机生成一个范围为 x ~ y 的实数
        randIndex = int(np.random.uniform(0, len(trainingSet)))
        testSet.append(trainingSet[randIndex]) 
        del(list(trainingSet)[randIndex]) # 从训练集中剔除测试集的数据
    trainMat = []
    trainClasses = []
    
    # 循环遍历训练集的所有文档,构建词向量
    for docIndex in trainingSet:  
        trainMat.append(setOfWords2Vec(vocabList, docList[docIndex])) # 训练集词向量
        trainClasses.append(classList[docIndex]) # 训练集每封邮件的分类
    p0V, p1V, pSpam = trainNB0(np.array(trainMat), np.array(trainClasses))
    errorCount = 0
                                  
    # 遍历测试集,对每封邮件进行分类
    for docIndex in testSet:        
        wordVector = setOfWords2Vec(vocabList, docList[docIndex])
        if classifyNB(np.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

函数 spamTest() 会输出在 10 封随机选择的电子邮件上的分类错误率。既然这些电子邮件是随机选择的,所以要想获得更好的估计错误率,应该将上述过程重复多次取平均值。

7. 实例 3:使用朴素贝叶斯分类器从个人广告中获取区域倾向

项目概述:广告商往往想知道关于一个人的一些特定人口统计信息,以便能更好地定向推销广告。

我们将分别从美国的两个城市中选取一些人,通过分析这些人发布的信息,来比较这两个城市的人们在广告用词上是否不同。如果结论确实不同,那么他们各自常用的词是哪些,从人们的用词当中,我们能否对不同城市的人所关心的内容有所了解。

👉 即查询两个 RSS 源的常用词是否相同。如果不同,他们各自的常用词是哪些

😂 这里的 RSS 源我们就随便选了两个国内的网站,可能不是广告网~

开发流程

  • 收集数据: 从 RSS 源收集内容,这里需要对 RSS 源构建一个接口
  • 准备数据: 将文本文件解析成词条向量
  • 分析数据: 检查词条确保解析的正确性
  • 训练算法: 使用我们之前建立的 trainNB0() 函数
  • 测试算法: 观察错误率,确保分类器可用。可以修改切分程序,以降低错误率,提高分类结果
  • 使用算法: 构建一个完整的程序,封装所有内容。给定两个 RSS 源,该程序会显示最常用的公共词

① 收集数据:导入 RSS 源

首先需要下载一个 RSS 阅读器:FeedParser

  • 打开 Anaconda Prompt 输入:pip install feedparser

  • 输入conda list 检查是否成功安装

使用 FeedParser

比如说访问人民网国际新闻的 RSS 源:

import feedparser

ny = feedparser.parse('http://www.people.com.cn/rss/world.xml')

可以构建一个类似于 spamTest() 的函数来对测试过程自动化:

#  朴素贝叶斯词袋模型
def bagOfWords2VecMN(vocabList, inputSet):
    returnVec = [0]*len(vocabList)
    for word in inputSet:
        if word in vocabList:
            returnVec[vocabList.index(word)] += 1
    return returnVec

def calcMostFreq(vocabList, fullText):
    """
    Desc:
        遍历词汇表中的每个词并统计它在文本中出现的次数
    Args:
        vocabList:词汇表
        fullText:文本(文档)
    Return:
        返回出现次数最高的30个单词
    """
    import operator
    freqDict = {}
    for token in vocabList:
        freqDict[token] = fullText.count(token)
    sortedFreq = sorted(freqDict.items(), key=operator.itemgetter(1), reverse=True)
    return sortedFreq[:30]

def localWords(feed1, feed0):
    import feedparser
    docList = []; classList = []; fullText = []
    minLen = min(len(feed1['entries']), len(feed0['entries']))
    for i in range(minLen):
        wordList = textParse(feed1['entries'][i]['summary'])
        docList.append(wordList)
        fullText.extend(wordList)
        classList.append(1) #NY is class 1
        wordList = textParse(feed0['entries'][i]['summary'])
        docList.append(wordList)
        fullText.extend(wordList)
        classList.append(0)
        
    vocabList = createVocabList(docList) #create vocabulary
    top30Words = calcMostFreq(vocabList, fullText)   # 30个高频词
    # 去除30个高频词
    for pairW in top30Words:
        if pairW[0] in vocabList: 
            vocabList.remove(pairW[0])
            
    trainingSet = range(2*minLen); testSet = []           #create test set
    # 随机构造测试集,训练集
    for i in range(20):
        randIndex = int(np.random.uniform(0, len(trainingSet)))
        testSet.append(trainingSet[randIndex])
        del(list(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(np.array(trainMat), np.array(trainClasses))
    errorCount = 0
    for docIndex in testSet:        # classify the remaining items
        wordVector = bagOfWords2VecMN(vocabList, docList[docIndex])
        if classifyNB(np.array(wordVector), p0V, p1V, pSpam) != classList[docIndex]:
            errorCount += 1
    print('the error rate is: ', float(errorCount)/len(testSet))
    
    return vocabList, p0V, p1V

🚩 函数 localWords() 使用了两个 RSS 源作为参数,RSS 源要在函数外导入,这样做的原因是 RSS 源会随时间而改变,重新加载 RSS 源就会得到新的数据。该函数的作用就是首先将一个RSS源作为类别 1,将另一个 RSS 源作为类别 2,然后从这两个 RSS 源的词汇表中随机构造测试集和训练集,对训练集中的词汇进行分类(类别 1 还是类别 0),然后对测试集进行分类,返回测试集上的错误率。

如果不移除高频词(注释掉移除高频词的三行代码),分类错误率将会显著上升。这说明词汇表中的一小部分却占据了所有文本用词的一大部分。

可以通过函数 caclMostFreq() 改变要移除的单词数目,然后观察错误率的变化情况。

② 分析数据:显示地域相关的用词

可以先对向量 pSF,pNY 进行排序,然后按照顺序将词打印出来:

# 显示最具表征性的词汇
# 最具表征性的词汇显示函数
def getTopWords(ny, sf):
    import operator
    vocabList, p0V, p1V = localWords(ny, sf)
    topNY = []; topSF = []
    for i in range(len(p0V)):
        if p0V[i] > -6.0: # 返回大于阈值的所有词
            topSF.append((vocabList[i], p0V[i]))
        if p1V[i] > -6.0: 
            topNY.append((vocabList[i], p1V[i]))
            
    sortedSF = sorted(topSF, key=lambda pair: pair[1], reverse=True)
    print("SF**SF**SF**SF**SF**SF**SF**SF**SF**SF**SF**SF**SF**SF**SF**SF**")
    for item in sortedSF:
        print(item[0])
        
    sortedNY = sorted(topNY, key=lambda pair: pair[1], reverse=True)
    print("NY**NY**NY**NY**NY**NY**NY**NY**NY**NY**NY**NY**NY**NY**NY**NY**")
    for item in sortedNY:
        print(item[0])

📚 References

posted @ 2023-01-10 17:51  RuoVea  阅读(31)  评论(0编辑  收藏  举报