tybrs commited on
Commit
217f149
1 Parent(s): fba584a

Update bias_auc.py

Browse files
Files changed (1) hide show
  1. 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]}')")