Selection rates in census dataset

from fairlearn.datasets import fetch_adult

from fairlearn.metrics import MetricFrame, selection_rate

data = fetch_adult()
X = data.data
y_true = (data.target == ">50K") * 1
sex = X["sex"]

selection_rates = MetricFrame(
    metrics=selection_rate, y_true=y_true, y_pred=y_true, sensitive_features=sex
)

fig = selection_rates.by_group.plot.bar(
    legend=False, rot=0, title="Fraction earning over $50,000"
)