十六、朴素贝叶斯

作者:Chris Albon

译者:飞龙

协议:CC BY-NC-SA 4.0

伯努利朴素贝叶斯

伯努利朴素贝叶斯分类器假设我们的所有特征都是二元的,它们仅有两个值(例如,已经是独热编码的标称分类特征)。

  1. # 加载库
  2. import numpy as np
  3. from sklearn.naive_bayes import BernoulliNB
  4. # 创建三个二元特征
  5. X = np.random.randint(2, size=(100, 3))
  6. # 创建二元目标向量
  7. y = np.random.randint(2, size=(100, 1)).ravel()
  8. # 查看前十个观测
  9. X[0:10]
  10. '''
  11. array([[1, 1, 1],
  12. [0, 1, 0],
  13. [1, 1, 1],
  14. [0, 0, 0],
  15. [1, 0, 1],
  16. [1, 1, 1],
  17. [0, 1, 1],
  18. [1, 1, 1],
  19. [1, 1, 1],
  20. [1, 1, 0]])
  21. '''
  22. # 创建伯努利朴素贝叶斯对象,带有每个类别的先验概率
  23. clf = BernoulliNB(class_prior=[0.25, 0.5])
  24. # 训练模型
  25. model = clf.fit(X, y)

校准预测概率

类别概率是机器学习模型中常见且有用的部分。 在 scikit-learn 中,大多数学习算法允许我们使用predict_proba来查看成员的类别预测概率。 例如,如果我们想要仅预测某个类,如果模型预测它们是该类的概率超过 90%,则这非常有用。 然而,一些模型,包括朴素贝叶斯分类器输出的概率,不基于现实世界。 也就是说,predict_proba可能预测,观测有 0.70 的机会成为某一类,而实际情况是它是 0.10 或 0.99。 特别是在朴素贝叶斯中,虽然不同目标类别的预测概率的排名是有效的,但是原始预测概率倾向于接近 0 和 1 的极值。

为了获得有意义的预测概率,我们需要进行所谓的校准。 在 scikit-learn 中,我们可以使用CalibratedClassifierCV类,使用 k-fold 交叉验证创建校准良好的预测概率。 在CalibratedClassifierCV中,训练集用于训练模型,测试集用于校准预测概率。返回的预测概率是 k 折的平均值。

  1. # 加载库
  2. from sklearn import datasets
  3. from sklearn.naive_bayes import GaussianNB
  4. from sklearn.calibration import CalibratedClassifierCV
  5. # 加载数据
  6. iris = datasets.load_iris()
  7. X = iris.data
  8. y = iris.target
  9. # 创建高斯朴素贝叶斯对象
  10. clf = GaussianNB()
  11. # 使用 sigmoid 校准创建校准的交叉验证
  12. clf_sigmoid = CalibratedClassifierCV(clf, cv=2, method='sigmoid')
  13. # 校准概率
  14. clf_sigmoid.fit(X, y)
  15. '''
  16. CalibratedClassifierCV(base_estimator=GaussianNB(priors=None), cv=2,
  17. method='sigmoid')
  18. '''
  19. # 创建新的观测
  20. new_observation = [[ 2.6, 2.6, 2.6, 0.4]]
  21. # 查看校准概率
  22. clf_sigmoid.predict_proba(new_observation)
  23. # array([[ 0.31859969, 0.63663466, 0.04476565]])

高斯朴素贝叶斯分类器

十六、朴素贝叶斯 - 图1

由于正态分布的假设,高斯朴素贝叶斯最适用于我们所有特征都是连续的情况。

  1. # 加载库
  2. from sklearn import datasets
  3. from sklearn.naive_bayes import GaussianNB
  4. # 加载数据
  5. iris = datasets.load_iris()
  6. X = iris.data
  7. y = iris.target
  8. # 创建高斯朴素贝叶斯对象,带有每个类别的先验概率
  9. clf = GaussianNB(priors=[0.25, 0.25, 0.5])
  10. # 训练模型
  11. model = clf.fit(X, y)
  12. # 创建新的观测
  13. new_observation = [[ 4, 4, 4, 0.4]]
  14. # 预测类别
  15. model.predict(new_observation)
  16. # array([1])

注意:来自高斯朴素贝叶斯的原始预测概率(使用predict_proba输出)未校准。 也就是说,他们不应该是可信的。 如果我们想要创建有用的预测概率,我们将需要使用等渗回归或相关方法来校准它们。

多项式逻辑回归

在多项逻辑回归(MLR)中,我们在 Recipe 15.1 中看到的逻辑函数被 softmax 函数替换:

十六、朴素贝叶斯 - 图2

其中 十六、朴素贝叶斯 - 图3 是第 十六、朴素贝叶斯 - 图4 个观测的目标值 十六、朴素贝叶斯 - 图5 是类 十六、朴素贝叶斯 - 图6 的概率,十六、朴素贝叶斯 - 图7 是类的总数。MLR 的一个实际优点是使用predict_proba方法预测的概率更可靠(即校准更好)。

  1. # 加载库
  2. from sklearn.linear_model import LogisticRegression
  3. from sklearn import datasets
  4. from sklearn.preprocessing import StandardScaler
  5. # 加载数据
  6. iris = datasets.load_iris()
  7. X = iris.data
  8. y = iris.target
  9. # 标准化特征
  10. scaler = StandardScaler()
  11. X_std = scaler.fit_transform(X)
  12. # 创建 OVR 逻辑回归对象
  13. clf = LogisticRegression(random_state=0, multi_class='multinomial', solver='newton-cg')
  14. # 训练模型
  15. model = clf.fit(X_std, y)
  16. # 创建新的观测
  17. new_observation = [[.5, .5, .5, .5]]
  18. # 预测类别
  19. model.predict(new_observation)
  20. # array([1])
  21. # 查看预测概率
  22. model.predict_proba(new_observation)
  23. # array([[ 0.01944996, 0.74469584, 0.2358542 ]])

多项式朴素贝叶斯分类器

多项式朴素贝叶斯的工作方式类似于高斯朴素贝叶斯,但假设这些特征是多项式分布的。 在实践中,这意味着当我们具有离散数据(例如,电影评级范围为 1 到 5)时,通常使用该分类器。

  1. # 加载库
  2. import numpy as np
  3. from sklearn.naive_bayes import MultinomialNB
  4. from sklearn.feature_extraction.text import CountVectorizer
  5. # 创建文本
  6. text_data = np.array(['I love Brazil. Brazil!',
  7. 'Brazil is best',
  8. 'Germany beats both'])
  9. # 创建词袋
  10. count = CountVectorizer()
  11. bag_of_words = count.fit_transform(text_data)
  12. # 创建特征矩阵
  13. X = bag_of_words.toarray()
  14. # 创建目标向量
  15. y = np.array([0,0,1])
  16. # 创建多项式朴素贝叶斯对象,带有每个类别的先验概率
  17. clf = MultinomialNB(class_prior=[0.25, 0.5])
  18. # 训练模型
  19. model = clf.fit(X, y)
  20. # 创建新的观测
  21. new_observation = [[0, 0, 0, 1, 0, 1, 0]]
  22. # 预测新观测的类别
  23. model.predict(new_observation)
  24. # array([0])

从零编写朴素贝叶斯分类器

朴素贝叶斯是一种简单的分类器,当只有少量观测可用时,这种分类器表现良好。 在本教程中,我们将从头开始创建一个高斯朴素贝叶斯分类器,并使用它来预测以前未见过的数据点的类别。本教程基于 Wikipedia 的朴素贝叶斯分类器页面上的示例,我已经用 Python 实现了它并调整了一些符号来改进解释。

  1. import pandas as pd
  2. import numpy as np

我们的数据集包含八个个体的数据。 我们将使用数据集构建一个分类器,该分类器接收个体的身高,体重和脚码,并输出其性别预测。

  1. # 创建空数据帧
  2. data = pd.DataFrame()
  3. # 创建我们的目标变量
  4. data['Gender'] = ['male','male','male','male','female','female','female','female']
  5. # 创建我们的特征变量
  6. data['Height'] = [6,5.92,5.58,5.92,5,5.5,5.42,5.75]
  7. data['Weight'] = [180,190,170,165,100,150,130,150]
  8. data['Foot_Size'] = [12,11,12,10,6,8,7,9]
  9. # 查看数据
  10. data
GenderHeightWeightFoot_Size
0male6.0018012
1male5.9219011
2male5.5817012
3male5.9216510
4female5.001006
5female5.501508
6female5.421307
7female5.751509

上面的数据集用于构造我们的分类器。 下面我们将创建一个新的个体,我们知道它的特征值,但不知道它的性别。我们的目标是预测它的性别。

  1. # 创建空数据帧
  2. person = pd.DataFrame()
  3. # 为这一行创建相同特征值
  4. person['Height'] = [6]
  5. person['Weight'] = [130]
  6. person['Foot_Size'] = [8]
  7. # 查看数据
  8. person
HeightWeightFoot_Size
061308

贝叶斯定理是一个着名的方程,它允许我们根据数据进行预测。 这是贝叶斯定理的经典版本:

十六、朴素贝叶斯 - 图8

这可能过于抽象,所以让我们替换一些变量以使其更具体。 在贝叶斯分类器中,给定数据的情况下,我们有兴趣找出观测的类别(例如男性或女性,垃圾邮件或非垃圾邮件):

十六、朴素贝叶斯 - 图9

其中:

  • 十六、朴素贝叶斯 - 图10 是特定类别(例如男性)
  • 十六、朴素贝叶斯 - 图11 是观测的数据
  • 十六、朴素贝叶斯 - 图12 称为后验
  • 十六、朴素贝叶斯 - 图13 叫做似然
  • 十六、朴素贝叶斯 - 图14 叫做先验
  • 十六、朴素贝叶斯 - 图15 叫做边缘概率

在贝叶斯分类器中,我们计算每个观测的每个类的后验(严格来说,我们只计算后验的分子,但现在忽略它)。 然后,基于后验值最大的类别对观测分类。 在我们的例子中,我们为观测预测两个可能的类别(例如男性和女性),因此我们将计算两个后验:一个用于男性,一个用于女性。

十六、朴素贝叶斯 - 图16

十六、朴素贝叶斯 - 图17

高斯朴素的贝叶斯可能是最受欢迎的贝叶斯分类器。 为了解释这个名称的含义,让我们看一下当我们应用两个类别(男性和女性)和三个特征变量(高度,重量和尺寸)时贝叶斯方程式的样子:

十六、朴素贝叶斯 - 图18

十六、朴素贝叶斯 - 图19

现在让我们解释一下上面的方程式:

  • 十六、朴素贝叶斯 - 图20 是先验概率。正如你所看到的,只是观测是男性的概率。 这只是数据集中的男性数量除以数据集中的总人数。
  • 十六、朴素贝叶斯 - 图21 是似然。注意我们已经解释了 十六、朴素贝叶斯 - 图22 所以它现在是数据集中的每个特征。“高斯”和“朴素”来自似然中的两个假设:
  • 如果你查看似然中的每项,你会注意到,我们假设每个特征彼此不相关。 也就是说,脚码与体重或身高等无关。这显然不是真的,而且是一个“朴素”的假设 - 因此称为“朴素贝叶斯”。
  • 其次,我们假设特征的值(例如女性的身体,女性的体重)通常是高斯分布的。这意味着 十六、朴素贝叶斯 - 图23 是通过将所需参数输入正态分布的概率密度函数来计算的:

十六、朴素贝叶斯 - 图24

  • 十六、朴素贝叶斯 - 图25 可能是贝叶斯方法中最令人困惑的部分之一。 在玩具示例(包括我们的)中,完全可以计算边际概率。 但是,在许多实际情况中,要找到边际概率的值极其困难或不可能(解释为什么超出了本教程的范围)。 对于我们的分类器来说,这并不像你想象的那么严重。 为什么? 因为我们不关心真正的后验值是什么,我们只关心哪个类具有最高的后验值。 并且因为边际概率对于所有类别都是相同的,(1)我们可以忽略分母,(2)只计算每个类的后验分子,(3)选择最大的分子。 也就是说,我们可以忽略后验分母,并仅根据后验分子的相对值进行预测。

好的! 理论结束。 现在让我们开始计算贝叶斯方程的所有不同部分。

先验可以是常数或概率分布。 在我们的例子中,这只是性别的概率。计算这很简单:

  1. # 男性数量
  2. n_male = data['Gender'][data['Gender'] == 'male'].count()
  3. # 女性数量
  4. n_female = data['Gender'][data['Gender'] == 'female'].count()
  5. # 总行数
  6. total_ppl = data['Gender'].count()
  7. # 男性比例
  8. P_male = n_male/total_ppl
  9. # 女性比例
  10. P_female = n_female/total_ppl

请记住,我们的似然中的每一项(例如 十六、朴素贝叶斯 - 图26)都可以看做正态的 PDF。 例如:

十六、朴素贝叶斯 - 图27

这意味着对于每个类别(例如女性)和特征(例如身高)组合,我们需要从数据计算方差和均值。Pandas 让这很容易:

  1. # 按性别分组数据,并计算每个特征的均值
  2. data_means = data.groupby('Gender').mean()
  3. # 查看值
  4. data_means
HeightWeightFoot_Size
Gender
female5.4175132.507.50
male5.8550176.2511.25
  1. # 按性别分组数据,并计算每个特征的方差
  2. data_variance = data.groupby('Gender').var()
  3. # 查看值
  4. data_variance
HeightWeightFoot_Size
Gender
female0.097225558.3333331.666667
male0.035033122.9166670.916667

现在我们可以创建我们需要的所有变量。 下面的代码可能看起来很复杂,但我们所做的,只是从上面两个表中的每个单元格中创建一个变量。

  1. # 男性的均值
  2. male_height_mean = data_means['Height'][data_variance.index == 'male'].values[0]
  3. male_weight_mean = data_means['Weight'][data_variance.index == 'male'].values[0]
  4. male_footsize_mean = data_means['Foot_Size'][data_variance.index == 'male'].values[0]
  5. # 男性的方差
  6. male_height_variance = data_variance['Height'][data_variance.index == 'male'].values[0]
  7. male_weight_variance = data_variance['Weight'][data_variance.index == 'male'].values[0]
  8. male_footsize_variance = data_variance['Foot_Size'][data_variance.index == 'male'].values[0]
  9. # Means for female
  10. female_height_mean = data_means['Height'][data_variance.index == 'female'].values[0]
  11. female_weight_mean = data_means['Weight'][data_variance.index == 'female'].values[0]
  12. female_footsize_mean = data_means['Foot_Size'][data_variance.index == 'female'].values[0]
  13. # Variance for female
  14. female_height_variance = data_variance['Height'][data_variance.index == 'female'].values[0]
  15. female_weight_variance = data_variance['Weight'][data_variance.index == 'female'].values[0]
  16. female_footsize_variance = data_variance['Foot_Size'][data_variance.index == 'female'].values[0]

最后,我们需要创建一个函数来计算每个似然项的概率密度(例如 十六、朴素贝叶斯 - 图28)。

  1. # 创建计算 p(x | y) 的函数
  2. def p_x_given_y(x, mean_y, variance_y):
  3. # 将参数输入到概率密度函数
  4. p = 1/(np.sqrt(2*np.pi*variance_y)) * np.exp((-(x-mean_y)**2)/(2*variance_y))
  5. # 返回 p
  6. return p

好的! 我们的贝叶斯分类器准备就绪。 请记住,既然我们可以忽略边际概率(分母),我们实际计算的是:

十六、朴素贝叶斯 - 图29

为此,我们只需要插入未分类个体(height = 6)的值,数据集的变量(例如女性身高的均值)和我们上面编写的函数(p_x_given_y):

  1. # 如果未分类的观测是男性的后验分子
  2. P_male * \
  3. p_x_given_y(person['Height'][0], male_height_mean, male_height_variance) * \
  4. p_x_given_y(person['Weight'][0], male_weight_mean, male_weight_variance) * \
  5. p_x_given_y(person['Foot_Size'][0], male_footsize_mean, male_footsize_variance)
  6. # 6.1970718438780782e-09
  1. # 如果未分类的观测是女性的后验分子
  2. P_female * \
  3. p_x_given_y(person['Height'][0], female_height_mean, female_height_variance) * \
  4. p_x_given_y(person['Weight'][0], female_weight_mean, female_weight_variance) * \
  5. p_x_given_y(person['Foot_Size'][0], female_footsize_mean, female_footsize_variance)
  6. # 0.00053779091836300176

因为女性的后验分子大于男性,所以我们预测这个人是女性。