Testing on a test set issue fix

arahusky · Dec 10, 2016 · 2cc681c · 2cc681c
1 parent 140340e
commit 2cc681c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/convolutional_autoencoder.py b/convolutional_autoencoder.py
@@ -311,7 +311,7 @@ def activator_binmasks(images, augmenter, parents, default):
                                                                           epoch_i, cost, end - start))
 
                 if batch_num % 100 == 0:
-                    test_inputs, test_targets = dataset.next_batch()
+                    test_inputs, test_targets = dataset.test_set
                     test_inputs = np.reshape(test_inputs, (-1, network.IMAGE_HEIGHT, network.IMAGE_WIDTH, 1))
                     test_targets = np.reshape(test_targets, (-1, network.IMAGE_HEIGHT, network.IMAGE_WIDTH, 1))
                     test_accuracy = sess.run(network.accuracy, feed_dict={network.inputs: test_inputs, network.targets: test_targets, network.is_training: False})