Little tweaks to beautify code in chapter 14

YaohanChen · Oct 6, 2016 · 6496cc6 · 6496cc6
1 parent cbb757f
commit 6496cc6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/14_recurrent_neural_networks.ipynb b/14_recurrent_neural_networks.ipynb
@@ -834,7 +834,7 @@
     "with tf.Session() as sess:\n",
     "    init.run()\n",
     "    for epoch in range(n_epochs):\n",
-    "        for iteration in range(len(mnist.test.labels)//batch_size):\n",
+    "        for iteration in range(mnist.train.num_examples // batch_size):\n",
     "            X_batch, y_batch = mnist.train.next_batch(batch_size)\n",
     "            X_batch = X_batch.reshape((-1, n_steps, n_inputs))\n",
     "            sess.run(training_op, feed_dict={X: X_batch, y: y_batch})\n",
@@ -1010,7 +1010,7 @@
     "with tf.Session() as sess:\n",
     "    init.run()\n",
     "    for epoch in range(n_epochs):\n",
-    "        for iteration in range(len(mnist.test.labels)//batch_size):\n",
+    "        for iteration in range(mnist.train.num_examples // batch_size):\n",
     "            X_batch, y_batch = mnist.train.next_batch(batch_size)\n",
     "            X_batch = X_batch.reshape((-1, n_steps, n_inputs))\n",
     "            sess.run(training_op, feed_dict={X: X_batch, y: y_batch})\n",
@@ -1837,7 +1837,7 @@
     "with tf.Session() as sess:\n",
     "    init.run()\n",
     "    for epoch in range(n_epochs):\n",
-    "        for iteration in range(len(mnist.test.labels)//batch_size):\n",
+    "        for iteration in range(mnist.train.num_examples // batch_size):\n",
     "            X_batch, y_batch = mnist.train.next_batch(batch_size)\n",
     "            X_batch = X_batch.reshape((batch_size, n_steps, n_inputs))\n",
     "            sess.run(training_op, feed_dict={X: X_batch, y: y_batch})\n",