Revisions

cli/output.py

@@ -8,11 +8,11 @@ def show_digit(pixels):
         print(" ".join("#" if p > 0.5 else "." for p in row))
 
 
-def explore(X_train, y_train):
+def explore(X_train, y_train, n=3):
     print("=" * 60)
     print("SAMPLE DIGITS")
     print("=" * 60)
-    for i in range(3):
+    for i in range(n):
         print(f"\nLabel: {y_train[i]}")
         show_digit(X_train[i])