kamangir commited on
Commit
29a052d
1 Parent(s): 3b27ab8

validating fashion_mnist train - kamangir/bolt#692

Browse files
image_classifier/__init__.py CHANGED
@@ -1,5 +1,5 @@
1
  name = "image_classifier"
2
 
3
- version = "1.1.69"
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(f"{output_path}/predictions.pyndarray", predictions):
 
 
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(f"{input_path}/predictions.pyndarray")
 
 
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)