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

Update bias_auc.py

Browse files
Files changed (1) hide show
  1. bias_auc.py +1 -1
bias_auc.py CHANGED
@@ -126,7 +126,7 @@ class BiasAUC(evaluate.Metric):
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]}')")
130
  score = np.nan
131
  return score
132
 
 
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: need negative and positive examples for both target subgroup '{args[-1]}' and background set")
130
  score = np.nan
131
  return score
132