validating fashion_mnist predict - kamangir/bolt#692

image_classifier/__init__.py

@@ -1,5 +1,5 @@
 name = "image_classifier"
 
-version = "1.1.121"
+version = "1.1.122"
 
 description = "fashion-mnist + hugging-face + awesome-bash-cli"

image_classifier/classes.py

@@ -74,7 +74,13 @@ class Image_Classifier(object):
 
         return True
 
-    def predict(self, test_images, test_labels, output_path="", page_count=-1):
+    def predict(
+        self,
+        test_images,
+        test_labels,
+        output_path="",
+        page_count=10,
+    ):
         logger.info(
             "image_classifier.predict({},{}){}".format(
                 string.pretty_shape_of_matrix(test_images),
@@ -428,7 +434,6 @@ class Image_Classifier(object):
             test_images,
             np.argmax(test_labels, axis=1),
             model_path,
-            page_count=10,
         )
 
     @property