sklearn朴素贝叶斯类库(naive_bayes)使用小结

转载 2018年04月17日 10:36:42

转载自:https://www.cnblogs.com/pinard/p/6074222.html
感谢原作者:刘建平Pinard

scikit-learn 朴素贝叶斯类库使用小结

  
  之前在朴素贝叶斯算法原理小结这篇文章中,对朴素贝叶斯分类算法的原理做了一个总结。这里我们就从实战的角度来看朴素贝叶斯类库。重点讲述scikit-learn朴素贝叶斯类库的使用要点和参数选择。

1. scikit-learn 朴素贝叶斯类库概述

  朴素贝叶斯是一类比较简单的算法,scikit-learn中朴素贝叶斯类库的使用也比较简单。相对于决策树,KNN之类的算法,朴素贝叶斯需要关注的参数是比较少的,这样也比较容易掌握。在scikit-learn中,一共有3个朴素贝叶斯的分类算法类。分别是GaussianNB,MultinomialNB和BernoulliNB。其中GaussianNB就是先验为高斯分布的朴素贝叶斯,MultinomialNB就是先验为多项式分布的朴素贝叶斯,而BernoulliNB就是先验为伯努利分布的朴素贝叶斯。

    这三个类适用的分类场景各不相同,一般来说,如果样本特征的分布大部分是连续值,使用GaussianNB会比较好。如果样本特征的分布大部分是多元离散值,使用MultinomialNB比较合适。而如果样本特征是二元离散值或者很稀疏的多元离散值,应该使用BernoulliNB。

2. GaussianNB类使用总结

    GaussianNB假设特征的先验概率为正态分布,即如下式:
    

P(Xj|Y=Ck)=12πσk2exp((xjμk)22σk2)

    其中Ck为Y的第k类类别。μkσk2为需要从训练集估计的值。

    GaussianNB会根据训练集求出μkσk2μk 为在样本类别Ck中,所有Xj(j=1,2,3...)的平均值。σk2为在样本类别Ck中,所有Xj(j=1,2,3...)的方差。

class sklearn.naive_bayes.GaussianNB(priors=None)

    GaussianNB类的主要参数仅有一个,即先验概率priors ,对应Y的各个类别的先验概率P(Y=Ck)。这个值默认不给出,如果不给出此时P(Y=Ck)=mkm。其中m为训练集样本总数量,mk为输出为第k类别的训练集样本数。如果给出的话就以priors 为准。

    在使用GaussianNB的fit方法拟合数据后,我们可以进行预测。此时预测有三种方法,包括predict,predict_log_proba和predict_proba

    predict方法就是我们最常用的预测方法,直接给出测试集的预测类别输出

    predict_proba则不同,它会给出测试集样本在各个类别上预测的概率。容易理解,predict_proba预测出的各个类别概率里的最大值对应的类别,也就是predict方法得到类别。

    predict_log_proba和predict_proba类似,它会给出测试集样本在各个类别上预测的概率的一个对数转化。转化后predict_log_proba预测出的各个类别对数概率里的最大值对应的类别,也就是predict方法得到类别。

下面给一个具体的例子,代码如下:

import numpy as np
X = np.array([[-1, -1], [-2, -1], [-3, -2], [1, 1], [2, 1], [3, 2]])#训练集
Y = np.array([1, 1, 1, 2, 2, 2])#每个点的类标签
from sklearn.naive_bayes import GaussianNB
clf = GaussianNB()

拟合数据

clf.fit(X, Y)#要先训练(调用fit方法)才能预测(调用predict方法)
print "==Predict result by predict=="
print(clf.predict([[-0.8, -1]]))#预测该点类别
print "==Predict result by predict_proba=="
print(clf.predict_proba([[-0.8, -1]]))
print "==Predict result by predict_log_proba=="
print(clf.predict_log_proba([[-0.8, -1]]))

结果如下:

==Predict result by predict==
[1]
==Predict result by predict_proba==
[[  9.99999949e-01   5.05653254e-08]]
==Predict result by predict_log_proba==
[[ -5.05653266e-08  -1.67999998e+01]]

    从上面的结果可以看出,测试样本[-0.8,-1]的类别预测为类别1。具体的测试样本[-0.8,-1]被预测为1的概率为9.99999949e-01 ,远远大于预测为2的概率5.05653254e-08。这也是为什么最终的预测结果为1的原因了。

    此外,GaussianNB一个重要的功能是有 partial_fit方法,这个方法的一般用在如果训练集数据量非常大,一次不能全部载入内存的时候。这时我们可以把训练集分成若干等分,重复调用partial_fit来一步步的学习训练集,非常方便。后面讲到的MultinomialNB和BernoulliNB也有类似的功能。

3. MultinomialNB类使用总结

    MultinomialNB假设特征的先验概率为多项式分布,即如下式:

P(Xj=xjl|Y=Ck)=xjl+λmk+nλ

    其中,P(Xj=xjl|Y=Ck)是第k个类别的第j维特征的第l个个取值条件概率。mk是训练集中输出为第k类的样本个数。λ 为一个大于0的常数,常常取为1,即拉普拉斯平滑。也可以取其他值。

class sklearn.naive_bayes.MultinomialNB(alpha=1.0, fit_prior=True,
class_prior=None)

    MultinomialNB参数比GaussianNB多,但是一共也只有仅仅3个。其中,参数alpha即为上面的常数λ,如果你没有特别的需要,用默认的1即可。如果发现拟合的不好,需要调优时,可以选择稍大于1或者稍小于1的数。布尔参数fit_prior表示是否要考虑先验概率,如果是false,则所有的样本类别输出都有相同的类别先验概率。否则可以自己用第三个参数class_prior输入先验概率,或者不输入第三个参数class_prior让MultinomialNB自己从训练集样本来计算先验概率,此时的先验概率为P(Y=Ck)=mk/m。其中m为训练集样本总数量,mk为输出为第k类别的训练集样本数。总结如下:

这里写图片描述

    在使用MultinomialNB的fit方法或者partial_fit方法拟合数据后,我们可以进行预测。此时预测有三种方法,包括predict,predict_log_proba和predict_proba。由于方法和GaussianNB完全一样,这里就不累述了。 

4. BernoulliNB类使用总结

    BernoulliNB假设特征的先验概率为二元伯努利分布,即如下式:

P(Xj=xjl|Y=Ck)=P(j|Y=Ck)xjl+(1P(j|Y=Ck))(1xjl)

    此时l只有两种取值。xjl只能取值0或者1。

class sklearn.naive_bayes.BernoulliNB(alpha=1.0, binarize=0.0,
fit_prior=True, class_prior=None)

    BernoulliNB一共有4个参数,其中3个参数的名字和意义和MultinomialNB完全相同。唯一增加的一个参数是binarize。这个参数主要是用来帮BernoulliNB处理二项分布的,可以是数值或者不输入。如果不输入,则BernoulliNB认为每个数据特征都已经是二元的。否则的话,小于binarize的会归为一类,大于binarize的会归为另外一类。

    在使用BernoulliNB的fit或者partial_fit方法拟合数据后,我们可以进行预测。此时预测有三种方法,包括predict,predict_log_proba和predict_proba。由于方法和GaussianNB完全一样,这里就不累述了。


若有错误,欢迎拍砖指正。。。

机器学习疯狂入门(2):朴素贝叶斯文本分类

-
  • 1970年01月01日 08:00

【机器学习】Naive Bayes 朴素贝叶斯

1.基本原理和步骤 Naive Bayes methods are a set of supervised learning algorithms based on applying Bay...
  • duxinyuhi
  • duxinyuhi
  • 2016-10-10 17:08:07
  • 1184

sklearn朴素贝叶斯类库使用小结

在scikit-learn中,提供了3中朴素贝叶斯分类算法:GaussianNB(高斯朴素贝叶斯)、MultinomialNB(多项式朴素贝叶斯)、BernoulliNB(伯努利朴素贝叶斯) sk...
  • kancy110
  • kancy110
  • 2017-05-26 09:18:24
  • 5900

sklearn.naive_bayes

#!/usr/bin/env python3 # -*- coding: utf-8 -*- """ Created on Sat Aug 5 12:07:15 2017@author: luoga...
  • luoganttcc
  • luoganttcc
  • 2017-08-05 12:17:31
  • 531

【机器学习 sklearn 】朴素贝叶斯naive_bayes

代码片段:# encoding: utf-8import sys reload(sys) sys.setdefaultencoding('utf-8')import pandas as pd impo...
  • u013421629
  • u013421629
  • 2017-11-24 15:44:44
  • 1365

Python机器学习库SKLearn分类算法之朴素贝叶斯

参考地址:http://scikit-learn.org/stable/modules/naive_bayes.html http://scikit-learn.org/stable/modules/...
  • cheng9981
  • cheng9981
  • 2017-04-28 21:38:15
  • 837

用Python开始机器学习(6:朴素贝叶斯分类器)

朴素贝叶斯 情感分类 python
  • lsldd
  • lsldd
  • 2014-11-27 14:52:04
  • 47189

sklearn 朴素贝叶斯 分类器

朴素贝叶斯朴素贝叶斯方法是一系列有监督学习的方法,这些方法基于对贝叶斯理论的应用,即简单(naive)的假设 每对特征之间都相互独立。给定类变量 y (这里一个样本仅属于一类) 和一个相互独立的特征向...
  • neuldp
  • neuldp
  • 2016-08-15 22:03:02
  • 2071

使用sklearn实现朴素贝叶斯文本分类

本文使用的python版本为3.4。 使用前需要安装numpy、matplotlib、scipy和scikitlearn,建议直接下载后安装,下载地址为:python包,选择相应的版本下载,注意不要...
  • shijing_0214
  • shijing_0214
  • 2016-07-20 22:23:15
  • 2727

sklearn的机器学习之路:朴素贝叶斯

1. 基础概念 什么是朴素贝叶斯:贝叶斯分类是一类分类算法的总称,这类算法均以贝叶斯定理为基础,故统称为贝叶斯分类。而朴素朴素贝叶斯分类是贝叶斯分类中最简单,也是常见的一种分类方法。 贝叶斯公式:...
  • qq_35044025
  • qq_35044025
  • 2018-02-13 16:48:10
  • 201
收藏助手
不良信息举报
您举报文章:sklearn朴素贝叶斯类库(naive_bayes)使用小结
举报原因:
原因补充:

(最多只允许输入30个字)