RNN5—LSTM文本分类模型实现3—计算图输入定义

本文详细介绍了一种基于LSTM的文本分类模型搭建过程,包括定义函数、获取类别数目、参数设置、构建模型各层、初始化及计算损失函数等关键步骤。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1.定义一个函数,实现文本分类的模型:

在这里插入图片描述

定义一个获得类别数目的api:

在这里插入图片描述

测试:

在这里插入图片描述
效果:
在这里插入图片描述
添加上两个变量:
在这里插入图片描述
在这里插入图片描述

2.将必要的参数从超参数集合里取出来,防止在后面经常用到:
在这里插入图片描述

定义输入、输出和drop out用到keep_prob:
在这里插入图片描述

定义能记录和保存现在训练到哪一步的global_step:

在这里插入图片描述
3.正式构建文本分类的模型:
定义embedding层的初始化函数:

在这里插入图片描述
定义LSTM层:

定义一个scale:

在这里插入图片描述
初始化lstm:

在这里插入图片描述

在这里插入图片描述
输出是一个三维的矩阵:

在这里插入图片描述
将最后一层是输出连接到全连接层上:
初始化全连接层:
在这里插入图片描述
计算损失函数:

在这里插入图片描述

计算train_op并返回计算的数值:
在这里插入图片描述
以上,LSTM文本分类模型就搭建完成了~

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值