tybrs commited on
Commit
1367a96
1 Parent(s): 803aa8f

Update bias_auc.py

Browse files
Files changed (1) hide show
  1. bias_auc.py +5 -5
bias_auc.py CHANGED
@@ -146,11 +146,11 @@ class BiasAUC(evaluate.Metric):
146
  mean_value = np.power(np.sum(metric_values)/(len(result) - 1), 1/power_value)
147
  result['Overall generalized mean'][f"{metric}"] = mean_value
148
  y_preds = [output[1] for output in outputs]
149
- try:
150
- result['Overall generalized mean']["Overall AUC"] = roc_auc_score(labels, y_preds)
151
- except ValueError:
152
- result['Overall generalized mean']["Overall AUC"] = np.nan
153
- return result
154
 
155
  def _compute(self, target, label, output, subgroups=None):
156
  if subgroups is None:
 
146
  mean_value = np.power(np.sum(metric_values)/(len(result) - 1), 1/power_value)
147
  result['Overall generalized mean'][f"{metric}"] = mean_value
148
  y_preds = [output[1] for output in outputs]
149
+ # try:
150
+ # result['Overall generalized mean']["Overall AUC"] = roc_auc_score(labels, y_preds)
151
+ # except ValueError:
152
+ # result['Overall generalized mean']["Overall AUC"] = np.nan
153
+ # return result
154
 
155
  def _compute(self, target, label, output, subgroups=None):
156
  if subgroups is None: