#122 · 主分类: 自然语言处理与文本处理

multi-class-text-classification-cnn-rnn

cnn embeddings kaggle lstm rnn tensorflow text-classification

将Kaggle旧金山犯罪描述分类为39个类别。在Tensorflow上使用CNN、RNN(GRU和LSTM)和词嵌入构建模型。

项目最后更新:06/06/26

GitHub Stars

602

Forks数量

258

贡献者数量

4

许可证

Apache-2.0

收录理由

多数文本分类教程止步于模型片段,这个仓库却把流程走完整了。它在Kaggle旧金山犯罪数据集上训练了一个TextCNN与GRU混合的网络,把自由文本的案情描述映射到39个犯罪类别,并演示了如何导出模型、对新数据做预测。训练和预测脚本、保存的词汇表、标签映射,都让它可以作为一套可复用的模板,适合想在自己标注文本上构建句子分类器的人。对于在短文本分类中比较CNN与RNN(GRU/LSTM)架构的团队,它也是不错的参考,因为代码展示了这两种方式如何在Keras里协同工作。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目