validating fashion_mnist train - kamangir/bolt#692

image_classifier/__init__.py

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

image_classifier/classes.py

@@ -121,13 +121,11 @@ class Image_Classifier(object):
                 cm,
                 self.class_names,
                 f"{output_path}/image_classifier/model/confusion_matrix.jpg",
-                {
-                    "header": [
-                        " | ".join(host.signature()),
-                        " | ".join(objects.signature()),
-                    ],
-                    "footer": self.signature(prediction_time),
-                },
+                header=[
+                    " | ".join(host.signature()),
+                    " | ".join(objects.signature()),
+                ],
+                footer=self.signature(prediction_time),
             ):
                 return False
 
@@ -145,14 +143,12 @@ class Image_Classifier(object):
                 distribution,
                 self.class_names,
                 f"{output_path}/image_classifier/model/label_distribution.jpg",
-                {
-                    "header": [
-                        " | ".join(host.signature()),
-                        " | ".join(objects.signature()),
-                    ],
-                    "footer": self.signature(prediction_time),
-                    "title": "distribution of test_labels",
-                },
+                header=[
+                    " | ".join(host.signature()),
+                    " | ".join(objects.signature()),
+                ],
+                footer=self.signature(prediction_time),
+                title="distribution of test_labels",
             ):
                 return False