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.45"
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
  )