lenoctambule/autoencoder
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(mnist_test.py): subplot invalid ncols

Browse Source
This commit is contained in:
Lenoctambule
2026-04-09 18:43:27 +02:00
parent 15865812d8
commit 075382cfb0
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/mnist_test.py
View File

@@ -89,12 +89,12 @@ def plot_random_reconstruction(autoencoder: AAutoencoder,
img_shape, img_shape,
y): y):
output, code = autoencoder.forward(example.flatten()) output, code = autoencoder.forward(example.flatten())
plt.subplot(1, 3, 1) plt.subplot(1, 2, 1)
plt.matshow( plt.matshow(
example.reshape(img_shape), example.reshape(img_shape),
fignum=False) fignum=False)
plt.title(f"Input ({y})") plt.title(f"Input ({y})")
plt.subplot(1, 3, 2) plt.subplot(1, 2, 2)
plt.matshow( plt.matshow(
output.reshape(img_shape), output.reshape(img_shape),
fignum=False) fignum=False)