Update function for balanced accuracy calculation.

MarcBS · Sep 13, 2019 · 09da3e4 · 09da3e4
1 parent a4a41f4
commit 09da3e4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/keras_wrapper/extra/evaluation.py b/keras_wrapper/extra/evaluation.py
@@ -247,8 +247,8 @@ def multiclass_metrics(pred_list, verbose, extra_vars, split):
     #accuracy_balanced = sklearn_metrics.accuracy_score(y_gt, y_pred, sample_weight=sample_weights, )
 
     # The following two lines should both provide the same measure (balanced accuracy)
-    #_, accuracy_balanced, _, _ = sklearn_metrics.precision_recall_fscore_support(y_gt, y_pred, average='macro')
-    accuracy_balanced = sklearn_metrics.balanced_accuracy_score(y_gt, y_pred)
+    _, accuracy_balanced, _, _ = sklearn_metrics.precision_recall_fscore_support(y_gt, y_pred, average='macro')
+    #accuracy_balanced = sklearn_metrics.balanced_accuracy_score(y_gt, y_pred)
 
     # Compute Precision, Recall and F1 score
     precision, recall, f1, _ = sklearn_metrics.precision_recall_fscore_support(y_gt, y_pred, average='micro')