基于BiGRU和贝叶斯分类器的文本分类
Text classification based on bidirectional GRU and Bayesian classifier作者机构:中北大学大数据学院山西太原030051 中北大学研究生院山西太原030051
出 版 物:《计算机工程与设计》 (Computer Engineering and Design)
年 卷 期:2020年第41卷第2期
页 面:381-385页
学科分类:081203[工学-计算机应用技术] 08[工学] 0835[工学-软件工程] 0812[工学-计算机科学与技术(可授工学、理学学位)]
基 金:山西省纪检信访大数据智能情报管理系统开发基金项目(201703D121025) 山西省回国留学人员科研基金项目(2017-093)
主 题:深度学习 文本分类 循环神经网络 GRU神经网络 贝叶斯分类器
摘 要:针对传统的循环神经网络模型在处理长期依赖问题时面临着梯度爆炸或者梯度消失的问题,且参数多训练模型时间长,提出一种基于双向GRU神经网络和贝叶斯分类器的文本分类方法。利用双向GRU神经网络提取文本特征,通过TF-IDF算法权重赋值,采用贝叶斯分类器判别分类,改进单向GRU对后文依赖性不足的缺点,减少参数,缩短模型的训练时间,提高文本分类效率。在两类文本数据上进行对比仿真实验,实验结果表明,该分类算法与传统的循环神经网络相比能够有效提高文本分类的效率和准确率。