NLP实战项目（3）——TextCNN结合BERT进行文本分类

1、数据集介绍

采用了清华NLP组提供的THUCNews新闻文本分类数据集的子集。其中训练集一共有 180000 条，验证集一共有 10000 条，测试集一共有 10000 条。其类别为 finance、realty、stocks、education、science、society、politics、sports、game、entertainment 十个类别。

2、模型结构

预训练模型下载地址：https://huggingface.co/google-bert

如果使用英文数据集则使用bert-base-uncased，如果是中文数据集则使用bert-base-chinese。

3、代码实现

文末免费获取数据集和源码压缩包。

训练曲线及测试精度：

进入下面公众号聊天窗口回复“CNNBERT文本分类”即可获取完整源码。

最后：

小编会不定期发布相关设计内容包括但不限于如下内容:信号处理、通信仿真、算法设计、matlab appdesigner，gui设计、simulink仿真......希望能帮到你！