赞
踩
工作中遇到了意图识别的分类问题,趁机从头开始学习了一下LSTM的结构以及代码编写,踩坑无数后终于成功的调通第一版深度学习代码。代码运用了tensorflow中的各种接口,相关的api请点 这里
我用的是query分词后term的索引id组成的query词向量
比如 “北京欢迎你”这个query,分词后为 北京_欢迎_你,对应的索引值为[1,5,7],即为一条训练数据。
由于这次分类任务只是简单的二分类,所以每一条数据都用一个二维向量表示它所在的类别,例如[1,0]
故,训练数据的shape和样例如下:
train_data的维度为[sample_size, max_len]
train_label的维度为[sample_size, class_num]
(max_len为query最大长度,class_num为类别个数,为了保持每条数据的长度都是max_len,需要进行0填充)
在最开始,将神经网络的参数导入到model中,如下:
# initialize parameters
self.keep_prob = config.keep_prob #dropout参数
se
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。