fix ROC fit_predict arguments (Trusted-AI#111)

Illia-Kryvoviaz · Oct 3, 2019 · 12ad4a5 · 12ad4a5
1 parent b972b59
commit 12ad4a5
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