validating fashion_mnist train - kamangir/bolt#689

image_classifier/__init__.py

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

image_classifier/classes.py

@@ -1,7 +1,11 @@
+import cv2
 from .plot import *
 from abcli import file
 from abcli import path
 from abcli import string
+from abcli.plugins import graphics
+from abcli.plugins import objects
+from abcli.tasks import host
 import numpy as np
 import matplotlib.pyplot as plt
 import tensorflow as tf
@@ -79,14 +83,17 @@ class Image_Classifier(object):
         prediction_time = (time.time() - prediction_time) / test_images.shape[0]
         logger.info(
             "image_classifier.predict(): {} / frame".format(
-                string.pretty_duration(prediction_time, include_ms=True)
+                string.pretty_duration(
+                    prediction_time,
+                    include_ms=True,
+                )
             )
         )
 
         if not output_path:
             return True
 
-        if not file.save("{}/predictions.pyndarray".format(output_path), predictions):
+        if not file.save(f"{output_path}/predictions.pyndarray", predictions):
             return False
 
         if test_labels is not None: