class BiasDetector:
def __init__(self, sensitive_attributes=['gender', 'age', 'race']):
self.sensitive_attributes = sensitive_attributes
def calculate_demographic_parity(self, predictions, sensitive_attr):
# 计算不同群体之间的预测结果差异
groups = np.unique(sensitive_attr)
group_rates = {}
for group in groups:
mask = sensitive_attr == group
group_rates[group] = predictions[mask].mean()