Skip to content

Commit aa0cae5

Browse files
committed
Adding plot for a deep autoencoder
1 parent 4af0372 commit aa0cae5

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed
17.2 KB
Loading

code/mnist_autoencoder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ def autoencoder_results(hidden_units):
2727
training_data, test_inputs, actual_test_results = \
2828
mnist_loader.load_data_nn()
2929
net = train_autoencoder(hidden_units, training_data)
30-
plot_test_results(net, test_inputs, actual_test_results)
30+
plot_test_results(net, test_inputs)
3131

3232
def train_autoencoder(hidden_units, training_data):
3333
"Return a trained autoencoder."
@@ -36,7 +36,7 @@ def train_autoencoder(hidden_units, training_data):
3636
net.SGD(autoencoder_training_data, 6, 10, 0.01, 0.05)
3737
return net
3838

39-
def plot_test_results(net, test_inputs, actual_test_results):
39+
def plot_test_results(net, test_inputs):
4040
"""
4141
Plot the results after passing the first ten test MNIST digits through
4242
the autoencoder ``net``."""

0 commit comments

Comments
 (0)