add wachter measures

oxfordinternetinstitute · Jul 25, 2024 · 0914ff7 · 0914ff7
1 parent 1506f26
commit 0914ff7
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/oxonfair/utils/group_metrics.py b/src/oxonfair/utils/group_metrics.py
@@ -218,6 +218,13 @@ def ge1(x):
                                                                                          - (TN+FP)/ge1(TTN+TFP)[:, np.newaxis]),
                                        'Absolute Difference of Proportion of Labels', total_metric=True)
 
+wachter_measures = {'pos_data_proportion': pos_data_proportion,
+                    'neg_data_proportion': neg_data_proportion,
+                    'diff_data_proportion': diff_data_proportion,
+                    'pos_pred_proportion': pos_pred_proportion,
+                    'neg_pred_proportion': neg_pred_proportion,
+                    'diff_pred_proportion': diff_pred_proportion}
+
 # directed bias amplification
 # see: https://proceedings.mlr.press/v139/wang21t/wang21t.pdf
 # warning do not use when enforcing fairness