Update bias_auc.py
Browse files- 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
|
| 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 |
|