`
yzmduncan
  • 浏览: 330374 次
  • 性别: Icon_minigender_1
  • 来自: 武汉
文章分类
社区版块
存档分类
最新评论

朴素贝叶斯在文本分类中的应用

阅读更多

       听过朴素贝叶斯的人,知道多项式朴素贝叶斯是神马,伯努利贝叶斯是神马吗?如果不知道,请继续读下去。

      其实所谓的“多项式”或“伯努利”,只不过是在求先验概率和条件概率时统计方法不一样,基本原理没变。

      贝叶斯分类算法(Naive Bayes)是一种基于统计的概率模型算法。在许多场合,朴素贝叶斯可以与决策树,SVM相媲美。它能应用到大数据中,方法简单且分类准确率高、速度快。但它基于属性之间的独立,在实际情况中经常是不成立的,因此分类准确率可能会下降。总的来说,NB是学习代价较低(理论简单,不像SVM和神经网络那么复杂),但效果较好的分类算法。

      贝叶斯分类算法基于托马斯贝叶斯发明的贝叶斯定理,他提出的贝叶斯定理对于现代概率论和数理统计的发展有重要的影响。

      贝叶斯定理:

           对于随机事件A和B:A发生的概率为P(A),B发生的概率为P(B),在B发生的情况下,A发生的概率为P(A|B)。A和B一起发生的联合概率为P(AB)。有:P(A|B) X P(B) = P(AB) = P(B|A) X P(A),则有:

P(A|B) = P(B|A)P(A) / P(B)

      文本分类(Text Categorization)是指计算机将一片文档归于预先给定的某一类或几类的过程。文本分类的特征提取过程是分词。目前比较好的中文分词器有中科院的ictclas,庖丁,IK等等。经过分词后,每个词就是一个特征。分词中可以自己配置停用词库,扩展词库等。特征选择有诸如TF-IDF,CHI等特征选择算法,就不在此赘述。

      朴素贝叶斯计算先验概率P(C)和条件概率P(X|C)的方法有两种:多项式模型伯努利模型。两者在计算的时候有两点差别:多项式会统计词频,而伯努利认为单词出现就记为1,没出现记为0,可以看到一个是基于词频,一个是基于文档频率;伯努利在分类时,将词库中的没有出现在待分类文本的词作为反方考虑

      在计算条件概率时,当待分类文本中的某个词没有出现在词库中时,概率为0,会导致很严重的问题,需要考虑拉普拉斯平滑(laplace smoothing):它是将所有词出现的次数+1,再进行统计。

      再一个问题就是概率太小而词数太多,会超double,用log将乘法转成加法

      多项式朴素贝叶斯算法伪代码如下:


 

       伯努利朴素贝叶斯算法伪代码如下:


 

多项式朴素贝叶斯代码

 

public class NaiveBayesManager {

    /**关键词索引 关键词-索引*/
    private Map<String, Integer> termIndex;
    /**类的索引 类名称-索引*/
    private Map<String, Integer> classIndex;
    /** 类名 */
    private List<String>         className;

    /**某类的文档中所有特征出现的总次数*/
    private int                  classTermsCount[];

    /**某类的文档中某特征出现的次数之和*/
    private int                  classKeyMap[][];

    /**类的个数*/
    private int                  numClasses      = 0;

    /**训练样本的所有特征(出现多次只算一个)*/
    private int                  vocabulary      = 0;

    /**训练样本的特征总次数*/
    private int                  totalTermsCount = 0;

    /** 建立类名和特征名的索引 */
    private void buildIndex(List<Corpus> orignCorpus) {
        classIndex = new HashMap<String, Integer>();
        termIndex = new HashMap<String, Integer>();
        className = new ArrayList<String>();
        Integer idTerm = new Integer(-1);
        Integer idClass = new Integer(-1);
        for (int i = 0; i < orignCorpus.size(); ++i) {
            Corpus corpus = orignCorpus.get(i);
            List<String> terms = corpus.getSegments();
            String label = corpus.getCat();
            if (!classIndex.containsKey(label)) {
                idClass++;
                classIndex.put(label, idClass);
                className.add(label);
            }
            for (String term : terms) {
                totalTermsCount++;
                if (!termIndex.containsKey(term)) {
                    idTerm++;
                    termIndex.put(term, idTerm);
                }
            }
        }
        vocabulary = termIndex.size();
        numClasses = classIndex.size();
    }

    /**
     * 训练
     * */
    public void startTraining(List<Corpus> orignCorpus) {
        buildIndex(orignCorpus);
        classTermsCount = new int[numClasses + 1];
        classKeyMap = new int[numClasses + 1][vocabulary + 1];
        for (int i = 0; i < orignCorpus.size(); ++i) {
            Corpus corpus = orignCorpus.get(i);
            List<String> terms = corpus.getSegments();
            String label = corpus.getCat();
            Integer labelIndex = classIndex.get(label);
            for (String term : terms) {
                Integer wordIndex = termIndex.get(term);
                classTermsCount[labelIndex]++;
                classKeyMap[labelIndex][wordIndex]++;
            }
        }
    }

    public String classify(List<String> terms) {
        int result = 0;
        double maxPro = Double.NEGATIVE_INFINITY;
        for (int cIndex = 0; cIndex < numClasses; ++cIndex) {
            double pro = Math.log10(getPreProbability(cIndex));
            for (String term : terms) {
                pro += Math.log10(getClassConditonalProbability(cIndex, term));
            }
            if (maxPro < pro) {
                maxPro = pro;
                result = cIndex;
            }
        }
        return className.get(result);
    }

    private double getPreProbability(int classIndex) {
        double ret = 0;
        int NC = classTermsCount[classIndex];
        int N = totalTermsCount;
        ret = 1.0 * NC / N;
        return ret;
    }

    private double getClassConditonalProbability(int classIndex, String term) {
        double ret = 0;
        int NCX = 0;
        int N = 0;
        int V = 0;

        Integer wordIndex = termIndex.get(term);
        if (wordIndex != null)
            NCX = classKeyMap[classIndex][wordIndex];

        N = classTermsCount[classIndex];

        V = vocabulary;

        ret = (NCX + 1.0) / (N + V); //laplace smoothing. 拉普拉斯平滑处理 
        return ret;
    }

    public Map<String, Integer> getTermIndex() {
        return termIndex;
    }

    public void setTermIndex(Map<String, Integer> termIndex) {
        this.termIndex = termIndex;
    }

    public Map<String, Integer> getClassIndex() {
        return classIndex;
    }

    public void setClassIndex(Map<String, Integer> classIndex) {
        this.classIndex = classIndex;
    }

    public List<String> getClassName() {
        return className;
    }

    public void setClassName(List<String> className) {
        this.className = className;
    }

    public int[] getClassTermsCount() {
        return classTermsCount;
    }

    public void setClassTermsCount(int[] classTermsCount) {
        this.classTermsCount = classTermsCount;
    }

    public int[][] getClassKeyMap() {
        return classKeyMap;
    }

    public void setClassKeyMap(int[][] classKeyMap) {
        this.classKeyMap = classKeyMap;
    }

    public int getNumClasses() {
        return numClasses;
    }

    public void setNumClasses(int numClasses) {
        this.numClasses = numClasses;
    }

    public int getVocabulary() {
        return vocabulary;
    }

    public void setVocabulary(int vocabulary) {
        this.vocabulary = vocabulary;
    }

    public int getTotalTermsCount() {
        return totalTermsCount;
    }

    public void setTotalTermsCount(int totalTermsCount) {
        this.totalTermsCount = totalTermsCount;
    }

    public static String getSplitword() {
        return splitWord;
    }

}

 

伯努利朴素贝叶斯代码

 

/**
 * @author zhongmin.yzm
 * 语料训练并载入内存
 * */
public class TrainingDataManager {

    /** 特征索引 */
    private Map<String, Integer> termIndex;
    /** 类索引 */
    private Map<String, Integer> classIndex;
    /** 索引-类名 */
    public List<String>          className;

    /**类的个数*/
    private int                  numClasses = 0;

    /**训练样本的所有特征(出现多次只算一个)*/
    private int                  vocabulary = 0;

    /**训练文本总数*/
    private int                  DocsNum    = 0;

    /**属于某类的文档个数*/
    private int[]                classDocs;

    /**类别c中包含属性 x的训练文本数量*/
    private int[][]              classKeyMap;

    /** 标志位: 分类时的优化 */
    private static boolean       flag[];

    private void buildIndex(List<List<String>> contents, List<String> labels) {
        classIndex = new HashMap<String, Integer>();
        termIndex = new HashMap<String, Integer>();
        className = new ArrayList<String>();
        Integer idTerm = new Integer(-1);
        Integer idClass = new Integer(-1);
        DocsNum = labels.size();
        for (int i = 0; i < DocsNum; ++i) {
            List<String> content = contents.get(i);
            String label = labels.get(i);
            if (!classIndex.containsKey(label)) {
                idClass++;
                classIndex.put(label, idClass);
                className.add(label);
            }
            for (String term : content) {
                if (!termIndex.containsKey(term)) {
                    idTerm++;
                    termIndex.put(term, idTerm);
                }
            }
        }
        vocabulary = termIndex.size();
        numClasses = classIndex.size();
    }

    public void startTraining(List<List<String>> contents, List<String> labels) {
        buildIndex(contents, labels);
        //去重
        List<List<Integer>> contentsIndex = new ArrayList<List<Integer>>();
        for (int i = 0; i < DocsNum; ++i) {
            List<Integer> contentIndex = new ArrayList<Integer>();
            List<String> content = contents.get(i);
            for (String str : content) {
                Integer wordIndex = termIndex.get(str);
                contentIndex.add(wordIndex);
            }
            Collections.sort(contentIndex);
            int num = contentIndex.size();
            List<Integer> tmp = new ArrayList<Integer>();
            for (int j = 0; j < num; ++j) {
                if (j == 0 || contentIndex.get(j - 1) != contentIndex.get(j)) {
                    tmp.add(contentIndex.get(j));
                }
            }
            contentsIndex.add(tmp);
        }
        //
        classDocs = new int[numClasses];
        classKeyMap = new int[numClasses][vocabulary];
        flag = new boolean[vocabulary];
        for (int i = 0; i < DocsNum; ++i) {
            List<Integer> content = contentsIndex.get(i);
            String label = labels.get(i);
            Integer labelIndex = classIndex.get(label);
            classDocs[labelIndex]++;
            for (Integer wordIndex : content) {
                classKeyMap[labelIndex][wordIndex]++;
            }
        }
    }

    /** 分类 时间复杂度 O(c*v) */
    public String classify(List<String> text) {
        double maxPro = Double.NEGATIVE_INFINITY;
        int resultIndex = 0;
        //标记待分类文本中哪些特征 属于 特征表
        for (int i = 0; i < vocabulary; ++i)
            flag[i] = false;
        for (String term : text) {
            Integer wordIndex = termIndex.get(term);
            if (wordIndex != null)
                flag[wordIndex] = true;
        }
        //对特征集中的每个特征: 若出现在待分类文本中,直接计算;否则作为反方参与
        for (int classIndex = 0; classIndex < numClasses; ++classIndex) {
            double pro = Math.log10(getPreProbability(classIndex));
            for (int wordIndex = 0; wordIndex < vocabulary; ++wordIndex) {
                if (flag[wordIndex])
                    pro += Math.log10(getClassConditionalProbability(classIndex, wordIndex));
                else
                    pro += Math.log10(1 - getClassConditionalProbability(classIndex, wordIndex));
            }
            if (maxPro < pro) {
                maxPro = pro;
                resultIndex = classIndex;
            }
        }
        return className.get(resultIndex);
    }

    /** 先验概率: 类C包含的文档数/总文档数 */
    private double getPreProbability(int classIndex) {
        double ret = 0.0;
        ret = 1.0 * classDocs[classIndex] / DocsNum;
        return ret;
    }

    /** 条件概率: 类C中包含关键字t的文档个数/类C包含的文档数 */
    private double getClassConditionalProbability(int classIndex, int termIndex) {
        int NCX = classKeyMap[classIndex][termIndex];
        int N = classDocs[classIndex];
        double ret = (NCX + 1.0) / (N + DocsNum);
        return ret;
    }

}

 

  • 大小: 6.2 KB
  • 大小: 16.2 KB
  • 大小: 19.8 KB
分享到:
评论

相关推荐

    基于朴素贝叶斯实现的文本分类

    在这个项目中,我们将深入探讨如何利用Python来实现朴素贝叶斯进行文本分类,并观察在某些情况下,分类准确率可以达到95%以上。 一、朴素贝叶斯理论基础 朴素贝叶斯分类器基于贝叶斯定理,该定理表述为:P(A|B) = P...

    朴素贝叶斯对于文本分类

    下面将详细介绍朴素贝叶斯在文本分类中的应用及其工作原理。 首先,我们要理解贝叶斯定理。贝叶斯定理是一个关于条件概率的公式,它表示在已知一些事件的情况下,另一事件发生的概率。用数学表达式表示为: P(A|B)...

    朴素贝叶斯文本分类器(新浪微博)

    朴素贝叶斯文本分类器是一种基于概率统计的机器学习算法,尤其在文本分类领域中广泛应用。这个特定的项目,"朴素贝叶斯文本分类器(新浪微博)"是针对新浪微博的数据进行情感分析,将微博内容分为积极和消极两类,...

    朴素贝叶斯算法文本分类JAVA实现

    朴素贝叶斯算法是一种基于概率的分类方法,它在机器学习领域被广泛应用,尤其是在文本分类中。该算法基于贝叶斯定理,假设特征之间相互独立,因此得名“朴素”。在Java环境下实现朴素贝叶斯分类器,我们可以分为以下...

    基于朴素贝叶斯算法的文本分类程序_Python

    朴素贝叶斯算法是一种在机器学习领域广泛应用的概率型分类方法,尤其在文本分类中表现出色。这个程序是用Python语言实现的,它利用朴素贝叶斯理论对文本数据进行分类。下面将详细介绍朴素贝叶斯算法及其在Python中的...

    朴素贝叶斯文本分类

    在实际应用中,朴素贝叶斯文本分类通常包含以下步骤: 1. **数据预处理**:包括分词、去除停用词、词干提取等,以减少噪声并标准化文本。 2. **特征提取**:利用TF-IDF计算每个单词的权重,形成特征向量。 3. **...

    基于朴素贝叶斯的垃圾邮件分类

    朴素贝叶斯算法是一种基于概率的分类方法,因其简单而有效,在处理文本分类问题,如垃圾邮件识别,中表现突出。本主题将深入探讨“基于朴素贝叶斯的垃圾邮件分类”这一技术。 朴素贝叶斯分类器基于贝叶斯定理,该...

    分布式朴素贝叶斯算法在文本分类中的应用.pdf

    朴素贝叶斯算法在文本分类任务中应用广泛,但是随着数据量的增大,算法的计算复杂度会增加,处理大规模数据集时效率低下。 在分布式朴素贝叶斯算法中,为了提高对大规模数据的处理能力,引入了MapReduce并行计算...

    朴素贝叶斯文本分类器(java实现)

    朴素贝叶斯文本分类器是一种广泛应用的机器学习算法,尤其在自然语言处理领域,用于将文本数据归类到预定义的类别中。本程序的Java实现深入探讨了这一概念,并提供了完整的工具集,包括源代码、实验报告、可执行程序...

    朴素贝叶斯 文本分类

    通过深入研究这个Java项目,我们可以掌握朴素贝叶斯文本分类的基本概念,并学习如何将理论应用于实践中。这个项目不仅提供了理论知识,还有实际的代码示例,是学习和应用朴素贝叶斯算法的好资源。

    朴素贝叶斯分类器算法

    在"朴素贝叶斯分类器算法"中,我们主要关注以下几个知识点: 1. **贝叶斯定理**:贝叶斯定理是统计学中的一个重要概念,用于在已知某个事件发生的条件下,计算另一个事件发生的概率。公式为:P(A|B) = [P(B|A) * P...

    高效朴素贝叶斯Web新闻文本分类模型的简易实现1

    朴素贝叶斯算法在文本分类中的应用是一种常见且有效的策略,尤其在处理大规模文本数据时。该算法基于贝叶斯定理,假设特征之间相互独立,每个特征对分类结果的影响等同,这使得计算简化,提高了处理速度。然而,由于...

    基于朴素贝叶斯分类器的文本分类算法

    朴素贝叶斯分类器是一种基于概率的机器学习方法,它在文本分类领域有着广泛的应用。该模型基于贝叶斯定理,并且通过“朴素”这一假设来简化计算,即假设特征之间是相互独立的。这一假设使得朴素贝叶斯分类器能够高效...

    基于朴素贝叶斯的文本分类算法.pdf

    最后,本文还提供了一些实践数据测试,展示了朴素贝叶斯算法在文本分类中的应用。 朴素贝叶斯算法是一种简单、快速、有效的文本分类方法,广泛应用于文本分类领域。但是,朴素贝叶斯算法也存在一些缺点,例如对数据...

    Python项目案例开发从入门到实战源代码第18章 机器学习案例——基于朴素贝叶斯算法的文本分类.rar

    在本项目案例中,我们将深入探讨如何使用Python进行机器学习,特别是通过朴素贝叶斯算法进行文本分类。朴素贝叶斯算法是一种基于概率的分类方法,它在处理文本数据时表现出色,因为其简单且效率高。这个项目是Python...

    朴素贝叶斯分类Iris数据

    朴素贝叶斯分类是一种基于概率的机器学习方法,它在数据分类中有着广泛的应用。该方法基于贝叶斯定理,假设特征之间相互独立,因此被称为“朴素”。在这个实例中,我们将探讨如何使用朴素贝叶斯分类器处理Iris数据集...

    机器学习(朴素贝叶斯)——文本分类

    朴素贝叶斯是一种基于概率的分类方法,它在机器学习领域有着广泛的应用,尤其是在文本分类中,如垃圾邮件检测。这种算法的理论基础是贝叶斯定理,它假设特征之间相互独立,这就是“朴素”一词的由来。下面我们将深入...

Global site tag (gtag.js) - Google Analytics