Update bias_auc.py
Browse files- bias_auc.py +1 -1
bias_auc.py
CHANGED
@@ -123,7 +123,7 @@ class BiasAUC(evaluate.Metric):
|
|
123 |
|
124 |
def _get_auc_score(self, gen_func, *args, **kwargs):
|
125 |
try:
|
126 |
-
y_trues, y_preds = zip(*gen_func(args))
|
127 |
score = roc_auc_score(y_trues, y_preds)
|
128 |
except ValueError:
|
129 |
print(f"Sample not sufficient for target clases '{args[-1]}' subgroup metric (need correct and incorrect predictions for '{args[-1]}')")
|
|
|
123 |
|
124 |
def _get_auc_score(self, gen_func, *args, **kwargs):
|
125 |
try:
|
126 |
+
y_trues, y_preds = zip(*gen_func(*args, **kwargs))
|
127 |
score = roc_auc_score(y_trues, y_preds)
|
128 |
except ValueError:
|
129 |
print(f"Sample not sufficient for target clases '{args[-1]}' subgroup metric (need correct and incorrect predictions for '{args[-1]}')")
|