基于TF-IDF的Naive Bayes文本分类器

简介：
本研究提出了一种利用TF-IDF改进词汇权重，并结合朴素贝叶斯算法进行文本分类的方法，有效提升了分类准确率。 我设计的基于文本的朴素贝叶斯分类器使用了TF/IDF平滑技术，并用C++实现。我的设计方案如下：首先为每个类别实例化4个映射，用于跟踪训练文件中的单词及其计数；另外还创建了4个映射来记录给定类别的每个词的概率。 通过读取训练数据（training.txt），我使用getline函数逐行处理文本，并对每一行的词汇进行解析。如果某个术语尚未出现在映射中，就会将其添加进去，反之则增加其计数。此外还有一个计算总字数的功能用于确定每种类别下的单词总数。 我还设计了一个computeProbability()函数来根据特定算法计算概率并存储至相应的概率图里；同时有一个分类（queue q）功能接受一个队列，并处理其中的每个词。该队列会将测试文件中某一行的所有词汇储存起来，然后计算出这一行在每种类别下的可能性值，最终选择最大值作为结果返回。