为了账号安全,请及时绑定邮箱和手机立即绑定

人工智能编程:循环神经网络RNN和长短时记忆模型LSTM的分析

标签:
Maya

人工智能编程:循环神经网络RNN和长短时记忆模型LSTM的分析

ht=tαnh(WihXt+bih+whhht-1+bhh)

在pytorch中我们使用nn.RNN()就可以创建出RNN神经网络,它有如下参数:

input_size表示输入xt的特征维度

hidden_size表示输出ht的特征维度,或者理解为隐藏层的神经元数

num_layers表示RNN网络的层数,默认是1层

nonlinearity表示非线性激活函数的选择,默认tanh,当然也可以选择relu

bias表示是否使用偏置,默认是Ture,使用batch_first这个参数是决定网络输入的维度顺序,默认是(seq,batch,feature)输入,seq表示序列长度,batch表示批量,feature表示特征维度,我们也可以将其修改为(batch,seq,feature),只用将这个参数设置为True dropout个参数接受一个0~1的数值,这个dropout层会在除了最后一层之外的其他输出层加上bidirectional默认的False,如果我们要是使用双向循环网络的话,那么我们就可以设置这个参数是True

点击查看更多内容
TA 点赞

若觉得本文不错,就分享一下吧!

评论

作者其他优质文章

正在加载中
  • 推荐
  • 评论
  • 收藏
  • 共同学习,写下你的评论
感谢您的支持,我会继续努力的~
扫码打赏,你说多少就多少
赞赏金额会直接到老师账户
支付方式
打开微信扫一扫,即可进行扫码打赏哦
今天注册有机会得

100积分直接送

付费专栏免费学

大额优惠券免费领

立即参与 放弃机会
意见反馈 帮助中心 APP下载
官方微信

举报

0/150
提交
取消