add ROC curve and scatter plots (mljar#81, mljar#158)

ajschumacher · Apr 14, 2021 · d46eea8 · d46eea8
1 parent f8d02ae
commit d46eea8
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/supervised/model_framework.py b/supervised/model_framework.py
@@ -416,12 +416,6 @@ def get_additional_metrics(self):
             else:
                 oof_preds = oof_predictions[prediction_cols]
 
-            # original_target = None
-            # if self._ml_task == BINARY_CLASSIFICATION:
-            #    original_target = self.preprocessings[0].inverse_categorical_target(
-            #        target["target"].values.ravel()
-            #    )
-
             sample_weight = None
             if "sample_weight" in oof_predictions.columns:
                 sample_weight = oof_predictions["sample_weight"]