Omit index when exporting prediction sets

Molmed · Nov 7, 2024 · ed5687b · ed5687b
1 parent ac7e713
commit ed5687b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/conformist/base_cop.py b/src/conformist/base_cop.py
@@ -135,7 +135,8 @@ def predict(self,
         self.create_output_dir(output_dir)
 
         # -- WRITE PREDICTIONS TO CSV
-        formatted_predictions.to_csv(f'{self.output_dir}/prediction_sets.csv')
+        formatted_predictions.to_csv(f'{self.output_dir}/prediction_sets.csv',
+                                     index=False)
 
         # -- GENERATE REPORTS --
         # Upset plot