当前位置:   article > 正文

LSTM文本分类(一)

lstm文本分类

LSTM文本分类(一)

工作中遇到了意图识别的分类问题,趁机从头开始学习了一下LSTM的结构以及代码编写,踩坑无数后终于成功的调通第一版深度学习代码。代码运用了tensorflow中的各种接口,相关的api请点 这里

1、训练数据的整理

我用的是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填充)

2、模型的构建

在最开始,将神经网络的参数导入到model中,如下:

# initialize parameters
self.keep_prob = config.keep_prob   #dropout参数
se
  • 1
  • 2
  • 3
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/码创造者/article/detail/1015707
推荐阅读
相关标签
  

闽ICP备14008679号