diff --git a/aif360/algorithms/inprocessing/meta_fair_classifier.py b/aif360/algorithms/inprocessing/meta_fair_classifier.py index 77ec4777..e31f7d2e 100644 --- a/aif360/algorithms/inprocessing/meta_fair_classifier.py +++ b/aif360/algorithms/inprocessing/meta_fair_classifier.py @@ -79,7 +79,7 @@ def predict(self, dataset): scores.append((t+1)/2) pred_dataset = dataset.copy() - pred_dataset.labels = np.array([predictions]) - pred_dataset.scores = np.array([scores]) + pred_dataset.labels = np.array([predictions]).T + pred_dataset.scores = np.array([scores]).T return pred_dataset diff --git a/aif360/algorithms/inprocessing/prejudice_remover.py b/aif360/algorithms/inprocessing/prejudice_remover.py index b7383c53..79c94b4e 100644 --- a/aif360/algorithms/inprocessing/prejudice_remover.py +++ b/aif360/algorithms/inprocessing/prejudice_remover.py @@ -148,8 +148,8 @@ def predict(self, dataset): all_sensitive_attributes, self.sensitive_attr, None) pred_dataset = dataset.copy() - pred_dataset.labels = predictions - pred_dataset.scores = scores + pred_dataset.labels = predictions[:, np.newaxis] + pred_dataset.scores = scores[:, np.newaxis] return pred_dataset