tensorflow学习笔记(九):单向、双向LSTM手写体(MNIST)识别



一、LSTM与Bi-LSTM简介

有关LSTM的介绍请参考:tensorflow学习笔记(八):LSTM手写体(MNIST)识别循环神经网络RNN详细推导LSTM(长短时间记忆模型)的详细推导

而Bi-LSTM与LSTM相似,只不过隐藏层的的block之间是双向传播的,适用于当前的信息不仅与前时刻的信息有关,还与后边的信息有关。如下图所示:
在这里插入图片描述

二、LSTM手写体识别

tensorflow学习笔记(八):LSTM手写体(MNIST)识别不同的是,这里定义的lstm单元采用的是 rnn.static_rnn,大家可以自行对比,有关两个函数的区别,在上一章给出了说明。

from __future__ import print_function

import tensorflow as tf
from tensorflow.contrib import rnn

# 导入MNIST data
from tensorflow.examples.tutorials.mnist import input_data
mnist = input_data.read_data_sets("/tmp/data/", one_hot=True)

# 训练参数
learning_rate = 0.001
training_steps = 1000
batch_size = 128
display_step = 200

# 网络参数
num_input = 28 # MNIST数据输入维度(img shape: 28*28)
timesteps = 28 # 步长
num_hidden = 128 # 隐藏层
num_classes = 10 # 分类 (0-9 个数字)

# 定义占位符
X = tf.placeholder("float", [None, timesteps, num_input])
Y = tf.placeholder("float", [None, num_classes])

# 定义权重
weights = {
    'out': tf.Variable(tf.truncated_normal([num_hidden, num_classes],stddev=0.1))
}
biases =
  • 0
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值