fixed label shapes to be (n, 1)

Illia-Kryvoviaz · Jun 7, 2023 · e06b11a · e06b11a
1 parent 1b78249
commit e06b11a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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