计算机工程

作者：彭俊利, 谷雨, 张震, 耿小航

单位：杭州电子科技大学 通信信息传输与融合技术国防重点学科实验室, 杭州 310000

关键词：单词贡献度,Word2Vec词向量,词嵌入,文档表示,文本分类

基金：国家自然科学基金（61673146）。

针对现有文档向量表示方法受噪声词语影响和重要词语语义不完整的问题，通过融合单词贡献度与Word2Vec词向量提出一种新的文档表示方法。应用数据集训练Word2Vec模型，计算数据集中词语的贡献度，同时设置贡献度阈值，提取贡献度大于该阈值的单词构建单词集合。在此基础上，寻找文档与集合中共同存在的单词，获取其词向量并融合单词贡献度生成文档向量。实验结果表明，该方法在搜狗中文文本语料库和复旦大学中文文本分类语料库上分类的平均准确率、召回率和F1值均优于TF-IDF、均值Word2Vec、PTF-IDF加权Word2Vec模型等传统方法，同时其对英文文本也能进行有效分类。

来源：2021年第4期

《计算机工程》期刊编辑部

查看计算机工程杂志2021年第4期