Change metric to show

enmanuelmag · Oct 13, 2022 · 6d716ad · 6d716ad
1 parent 35dc07b
commit 6d716ad
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/iaflow/__init__.py b/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}')

diff --git a/setup.py b/setup.py
@@ -6,7 +6,7 @@
 
 setuptools.setup(
     name="iaflow",
-    version='2.1.12',
+    version='2.1.13',
     author="Enmanuel Magallanes Pinargote",
     author_email="[email protected]",
     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.",