kamangir
commited on
Commit
•
67dfd4a
1
Parent(s):
8dd0f73
validating image_classifier - kamangir/bolt#689
Browse files
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.49"
|
4 |
|
5 |
description = "fashion-mnist + hugging-face + awesome-bash-cli"
|
image_classifier/fashion_mnist/ingest.py
CHANGED
@@ -30,7 +30,7 @@ def ingest(output_path):
|
|
30 |
"train_images,train_labels,test_images,test_labels".split(","),
|
31 |
[train_images, train_labels, test_images, test_labels],
|
32 |
):
|
33 |
-
if file.save(os.path.join(output_path, name), thing):
|
34 |
logger.info(f"ingested {name}: {string.pretty_size_of_matrix(thing)}")
|
35 |
else:
|
36 |
success = False
|
@@ -47,7 +47,7 @@ def ingest(output_path):
|
|
47 |
"Bag",
|
48 |
"Ankle boot",
|
49 |
]
|
50 |
-
if file.save_json(os.path.join(output_path, "class_names"), class_names):
|
51 |
logger.info(
|
52 |
f"ingested {len(class_names)} class name(s): {', '.join(class_names)}"
|
53 |
)
|
|
|
30 |
"train_images,train_labels,test_images,test_labels".split(","),
|
31 |
[train_images, train_labels, test_images, test_labels],
|
32 |
):
|
33 |
+
if file.save(os.path.join(output_path, f"{name}.pyndarray"), thing):
|
34 |
logger.info(f"ingested {name}: {string.pretty_size_of_matrix(thing)}")
|
35 |
else:
|
36 |
success = False
|
|
|
47 |
"Bag",
|
48 |
"Ankle boot",
|
49 |
]
|
50 |
+
if file.save_json(os.path.join(output_path, "class_names.json"), class_names):
|
51 |
logger.info(
|
52 |
f"ingested {len(class_names)} class name(s): {', '.join(class_names)}"
|
53 |
)
|