kamangir commited on
Commit
5dbe5c2
1 Parent(s): ba62492

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.67"
4
 
5
  description = "fashion-mnist + hugging-face + awesome-bash-cli"
 
1
  name = "image_classifier"
2
 
3
+ version = "1.1.68"
4
 
5
  description = "fashion-mnist + hugging-face + awesome-bash-cli"
image_classifier/classes.py CHANGED
@@ -110,13 +110,15 @@ class Image_Classifier(object):
110
  cm = cm / np.sum(cm, axis=1)[:, np.newaxis]
111
  logger.debug("confusion_matrix: {}".format(cm))
112
 
113
- if not file.save("{}/confusion_matrix.pyndarray".format(output_path), cm):
 
 
114
  return False
115
 
116
  if not graphics.render_confusion_matrix(
117
  cm,
118
  self.class_names,
119
- "{}/Data/0/info.jpg".format(output_path),
120
  {
121
  "header": [
122
  " | ".join(host.signature()),
@@ -140,7 +142,7 @@ class Image_Classifier(object):
140
  if not graphics.render_distribution(
141
  distribution,
142
  self.class_names,
143
- "{}/Data/1/info.jpg".format(output_path),
144
  {
145
  "header": [
146
  " | ".join(host.signature()),
 
110
  cm = cm / np.sum(cm, axis=1)[:, np.newaxis]
111
  logger.debug("confusion_matrix: {}".format(cm))
112
 
113
+ if not file.save(
114
+ f"{output_path}/image_classifier/model/confusion_matrix.pyndarray", cm
115
+ ):
116
  return False
117
 
118
  if not graphics.render_confusion_matrix(
119
  cm,
120
  self.class_names,
121
+ f"{output_path}/image_classifier/model/confusion_matrix.jpg",
122
  {
123
  "header": [
124
  " | ".join(host.signature()),
 
142
  if not graphics.render_distribution(
143
  distribution,
144
  self.class_names,
145
+ f"{output_path}/image_classifier/model/label_distribution.jpg",
146
  {
147
  "header": [
148
  " | ".join(host.signature()),