annyfangfang的博客

原创 使用LSTM训练分类模型（kreas+tensorflow）

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