HanLP-朴素贝叶斯分类预测缺陷

 

文章整理自 baiziyu 的知乎专栏，感兴趣的朋友可以去关注下这位大神的专栏，很多关于自然语言处理的文章写的很不错。昨天看到他的分享的两篇关于朴素贝叶斯分类预测的文章，整理了一下分享给给大家，文章已做部分修改！

 

朴素贝叶斯分类时，最好取对数变相乘为相加，防止预测结果溢出。可能出现的badcase就是明明训练语料X类目下没有词语t，而系统就将文本预测为X类目。解决方法就时改相乘为取对数相加。HanLP的朴素贝叶斯分类计算没有用对数相加的方法，而是直接用的概率相乘，很有可能溢出。

 

对上述内容做一些更正，HanLP的朴素贝叶斯是按照概率取对数相加做的。

看一下下边的代码

 

 @Override

    public double[] categorize(Document document) throws IllegalArgumentException, IllegalStateException

    {

        Integer category;

        Integer feature;

        Integer occurrences;

        Double logprob;

 

        double[] predictionScores = new double[model.catalog.length];

        for (Map.Entry<Integer, Double> entry1 : model.logPriors.entrySet())

        {

            category = entry1.getKey();

            logprob = entry1.getValue(); //用类目的对数似然初始化概率

 

            //对文档中的每个特征

            for (Map.Entry<Integer, int[]> entry2 : document.tfMap.entrySet())

            {

                feature = entry2.getKey();

 

                if (!model.logLikelihoods.containsKey(feature))

                {

                    continue; //如果在模型中找不到就跳过了

                }

 

                occurrences = entry2.getValue()[0]; //获取其在文档中的频次

 

                logprob += occurrences * model.logLikelihoods.get(feature).get(category); //将对数似然乘上频次

            }

            predictionScores[category] = logprob;

        }

 

        if (configProbabilityEnabled) MathUtility.normalizeExp(predictionScores);

        return predictionScores;

    }

 

这么看来，之前遇到的下边的这个badcase就还要再分析

[1] 化验指标一变化患者就六神无主，看医生怎么讲解

核心词：患者 看医生

这里“患者”和“看医生”两个词都没在“艺术”类训练语料中出现，但是预测概率最大的反倒是“艺术”。

由于用PyHanLP没法看到预测概率的计算过程，所以还是把Python的分类预测代码改为Java代码调式看一下。今天移植了预处理，资源加载，人工干预部分的代码，明天把剩余预测部分移植为Java再来看这个badcase。这就是朴素贝叶斯的优势，分析起来非常清晰容易。不过从PyHanLP的预测输出概率值来看，不太像是取了对数相加得到的，因为都是0-1之间的数值，这一看就是概率值。