kamangir commited on
Commit ·
29a052d
1
Parent(s): 3b27ab8
validating fashion_mnist train - kamangir/bolt#692
Browse files- image_classifier/__init__.py +1 -1
- image_classifier/classes.py +3 -1
- image_classifier/funcs.py +3 -1
image_classifier/__init__.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
name = "image_classifier"
|
| 2 |
|
| 3 |
-
version = "1.1.
|
| 4 |
|
| 5 |
description = "fashion-mnist + hugging-face + awesome-bash-cli"
|
|
|
|
| 1 |
name = "image_classifier"
|
| 2 |
|
| 3 |
+
version = "1.1.71"
|
| 4 |
|
| 5 |
description = "fashion-mnist + hugging-face + awesome-bash-cli"
|
image_classifier/classes.py
CHANGED
|
@@ -93,7 +93,9 @@ class Image_Classifier(object):
|
|
| 93 |
if not output_path:
|
| 94 |
return True
|
| 95 |
|
| 96 |
-
if not file.save(
|
|
|
|
|
|
|
| 97 |
return False
|
| 98 |
|
| 99 |
if test_labels is not None:
|
|
|
|
| 93 |
if not output_path:
|
| 94 |
return True
|
| 95 |
|
| 96 |
+
if not file.save(
|
| 97 |
+
f"{output_path}/image_classifier/predictions.pyndarray", predictions
|
| 98 |
+
):
|
| 99 |
return False
|
| 100 |
|
| 101 |
if test_labels is not None:
|
image_classifier/funcs.py
CHANGED
|
@@ -23,7 +23,9 @@ def eval(input_path, output_path):
|
|
| 23 |
",".join([str(value) for value in ground_truth[:10]] + ["..."]),
|
| 24 |
)
|
| 25 |
)
|
| 26 |
-
success, predictions = file.load(
|
|
|
|
|
|
|
| 27 |
|
| 28 |
if success:
|
| 29 |
predictions = np.argmax(predictions, axis=1).astype(np.uint8)
|
|
|
|
| 23 |
",".join([str(value) for value in ground_truth[:10]] + ["..."]),
|
| 24 |
)
|
| 25 |
)
|
| 26 |
+
success, predictions = file.load(
|
| 27 |
+
f"{input_path}/image_classifier/predictions.pyndarray"
|
| 28 |
+
)
|
| 29 |
|
| 30 |
if success:
|
| 31 |
predictions = np.argmax(predictions, axis=1).astype(np.uint8)
|