NLP入门竞赛，搜狗新闻文本分类

搜狗新闻文本分类竞赛

使用keras框架通过构建CNN+BiGRU网络实现在搜狗新闻文本数据集上91+的门竞准确率。

◆◆数据集来源◆◆

该数据集来自若干新闻站点2012年6月—7月期间国内，赛搜国际，狗新721re网址最新获取域名体育，闻文社会，本分娱乐等18个频道的门竞新闻数据。

根据新闻正文内容分析新闻的赛搜类别数据集官网链接：

 http://www.sogou.com/labs/resource/tce.php.

 该数据集样例格式如下所示：

在 FlyAI竞赛平台上 提供了超详细的参考代码，我们可以通过参加搜狗新闻文本分类预测练习赛进行进一步学习和优化。狗新

◆◆代码实现◆◆1.1、闻文算法流程及实现

算法流程主要分为以下四个部分进行介绍：

1.数据加载

2.构建网络

3.模型训练

1.数据加载

对每条新闻数据的本分721re网址最新获取域名读取和处理是在processor.py文件中完成。

具体实现如下：

2.构建网络

由于是门竞搜狗新闻文本类数据，这里我们可以使用一维卷积Conv1D + BiGRU来构建网络，赛搜网络结构如下所示：

运行summary()方法后输出的狗新网络结构如下图：

3.模型训练

这里我们设置了epoch为5，batch为128，闻文采用adam优化器来训练网络，本分EarlyStopping可以加速调参过程。然后通过调用FlyAI提供的train_log方法可以在训练过程中实时的看到训练集和验证集的准确率及损失变化曲线。

1.2.最终结果

通过使用自定义CNN网络结构+双向GRU网络的方法，在epoch为10，batch为128的条件下使用adam优化器下不断优化模型参数，使用early_stopping规则在model训练达到early_stopping条件时提前终止训练提高model优化效率，最终模型在测试集的准确率达到91+。

END

更多详情可

扫描下方二维码或添加AINLP微信号(id: AINLP2)

进群，和大神一起打比赛

备注“公众号名+flyai”

免费p40 GPU时长等你来领

点击“阅读原文”直达竞赛