fix ROC fit_predict arguments (#111)

Trusted-AI · Oct 3, 2019 · 53cf097 · 53cf097
1 parent 915e016
commit 53cf097
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/aif360/algorithms/postprocessing/reject_option_classification.py b/aif360/algorithms/postprocessing/reject_option_classification.py
@@ -211,9 +211,9 @@ def predict(self, dataset):
 
         return dataset_new
 
-    def fit_predict(self, dataset):
+    def fit_predict(self, dataset_true, dataset_pred):
         """fit and predict methods sequentially."""
-        return self.fit().predict(dataset)
+        return self.fit(dataset_true, dataset_pred).predict(dataset_pred)
 
 # Function to obtain the pareto frontier
 def _get_pareto_frontier(costs, return_mask = True):  # <- Fastest for many points