utils/fairness_utils.py

from fairlearn.metrics import MetricFrame, selection_rate, accuracy_score_group_min

def fairness_by_group(y_true, y_pred, sensitive_features):
    metrics = {
        "accuracy": accuracy_score_group_min,
        "selection_rate": selection_rate,
    }
    frame = MetricFrame(metrics=metrics, y_true=y_true, y_pred=y_pred, sensitive_features=sensitive_features)
    return frame.by_group