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