#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里协同工作。
相关文章
暂无针对该项目的文章。
如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。