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