add chosen column

cavalab · Dec 21, 2023 · f02616a · f02616a
1 parent e7da3d0
commit f02616a
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/fomo/estimator.py b/fomo/estimator.py
@@ -369,10 +369,14 @@ def _get_signed_F(self, F=None):
 
     def get_pareto_points(self):
         """Return a Pandas dataframe of the Pareto archive points"""
+        F = self._get_signed_F() 
+        I = self.I_
         archive = pd.DataFrame(
-            self._get_signed_F(), 
+            F,
             columns=self.accuracy_metrics_ + self.fairness_metrics_
         )
+        chosen = [f==F[I] for f in F]
+        archive['chosen'] = chosen
         return archive
 
 class FomoClassifier(FomoEstimator, ClassifierMixin, BaseEstimator):