Tensorflow之RNN,LSTM
Tensorflow之RNN,LSTM #!/usr/bin/env python2 # -*- coding: utf-8 -*- """ tensorflow之RNN 循环神经网络做手写数据集分类 """ import tensorflow as tf from tensorflow.examples.tutorials.mnist import input_data #设置随机数来比较两种计算结果 tf.set_random_seed(1) #导入手写数据集 mnist = input_data.read_data_sets('MNIST_data', one_hot=True) #设置参数 lr = 0.001 training_iters = 100000 batch_size = 128 n_inputs = 28 # MNIST 输入为图片(img shape: 28*28)对应到图片像素的一行 n_steps = 28 # time steps 对应到图片有多少列 n_hidden_units = 128 # 隐藏层神经元个数 n_classes = 10 # MNIST分类结果为10 #定义权重 weights = { #(28,128) 'in': tf.Variable(tf.random_normal([n_inputs, n_hidden_units]))