使用LSTM训练分类模型(kreas+tensorflow)
github链接:https://github.com/fangxiaozhu/Classification-model
需求描述:基于构建简单的多分类模型需求,使用的是keras深度学习库实现的,实现代码简单可用,模型优点是训练速度快,准确率达97%,适合粗粒度分类。
实现步骤:1、预处理数据:包括构建自定义词典、分词、去停用词等,处理成类似这样的训练数据格式
2、预训练word2vec模型:这里使用的是26万相关业务语料训练,保存成query.all.split.pkl模型,然后用pickle
原创
2020-09-25 14:24:08 ·
3297 阅读 ·
0 评论