Change metric to show

Author: enmanuelmag

iaflow/__init__.py

@@ -445,12 +445,12 @@ def plot_history(self, history, path):
     }
 
     for idx in range(len(train_loss)):
-      cur_val_loss = val_loss[idx]
-      if cur_val_loss < results['val_loss']:
+      cur_val_acc = val_acc[idx]
+      if cur_val_acc < results['val_acc']:
         results['train_loss'] = train_loss[idx]
         results['train_acc'] = train_acc[idx]
-        results['val_loss'] = cur_val_loss
-        results['val_acc'] = val_acc[idx]
+        results['val_loss'] = val_loss[idx]
+        results['val_acc'] = cur_val_acc
 
     for key, value in results.items():
       print(f'{key}: {value}')

setup.py

@@ -6,7 +6,7 @@
 
 setuptools.setup(
     name="iaflow",
-    version='2.1.12',
+    version='2.1.13',
     author="Enmanuel Magallanes Pinargote",
     author_email="enmanuelmag@cardor.dev",
     description="This library help to create models with identifiers, checkpoints, logs and metadata automatically, in order to make the training process more efficient and traceable.",