fix(mnist_test.py): subplot invalid ncols

2026-04-09 18:43:27 +02:00
parent 15865812d8
commit 075382cfb0
1 changed files with 2 additions and 2 deletions
--- a/examples/mnist_test.py
+++ b/examples/mnist_test.py
@@ -89,12 +89,12 @@ def plot_random_reconstruction(autoencoder: AAutoencoder,
                               img_shape,
                               y):
    output, code = autoencoder.forward(example.flatten())
-    plt.subplot(1, 3, 1)
+    plt.subplot(1, 2, 1)
    plt.matshow(
        example.reshape(img_shape),
        fignum=False)
    plt.title(f"Input ({y})")
-    plt.subplot(1, 3, 2)
+    plt.subplot(1, 2, 2)
    plt.matshow(
        output.reshape(img_shape),
        fignum=False)