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)
|