Optionally return validation run

Molmed · Nov 24, 2024 · 25b6b79 · 25b6b79
1 parent 87b2933
commit 25b6b79
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/conformist/base_cop.py b/src/conformist/base_cop.py
@@ -145,6 +145,7 @@ def predict(self,
             self.output_dir,
             upset_plot_color)
 
+        vr = None
         if validate:
             vr = ValidationRun(
                     self.val_idx,
@@ -170,7 +171,7 @@ def predict(self,
             df = pd.DataFrame(stats, index=[0])
             df.T.to_csv(f'{self.output_dir}/summary.csv', header=False)
 
-        return formatted_predictions
+        return formatted_predictions, vr
 
     def prediction_set_to_text(self, prediction_set, display_classes=None):
         class_names = self.class_names