updated cli output

Author: max-mauermann

birdnet_analyzer/train/utils.py

@@ -544,8 +544,8 @@ def run_trial(self, trial, *args, **kwargs):
 
     print("...Done.", flush=True)
 
-    # Get best validation metrics based on AUPRC instead of loss for more reliable results with imbalanced data
-    best_epoch = np.argmax(history.history["val_AUPRC"])
+    # Get best validation metrics based on loss
+    best_epoch = np.argmin(history.history["val_loss"])
     best_val_auprc = history.history["val_AUPRC"][best_epoch]
     best_val_auroc = history.history["val_AUROC"][best_epoch]
     best_val_loss = history.history["val_loss"][best_epoch]