深度学习实战3-文本卷积神经网络(TextCNN)新闻文本分类
文章目录一、前期工作1. 设置GPU2. 导入预处理词库类二、导入预处理词库类三、参数设定四、创建模型五、训练模型函数六、测试模型函数七、训练模型与预测今天给大家带来一个简单的中文新闻分类模型,利用TextCNN模型进行训练,TextCNN的主要流程是:获取文本的局部特征:通过不同的卷积核尺寸来提取文本的N-Gram信息,然后通过最大池化操作来突出各个卷积操作提取的最关键信息,拼接后通过全连接层对特征进行组合,最后通过交叉熵损失函数来训练模型。注:N-Gram是大词汇连续语音识别中