Tweak upset plot appearance

Molmed · Nov 21, 2024 · 5d283ed · 5d283ed
1 parent ba688e7
commit 5d283ed
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/conformist/prediction_dataset.py b/src/conformist/prediction_dataset.py
@@ -526,6 +526,7 @@ def visualize_prediction_stripplot(self,
 
     def visualize_model_sets(self, min_softmax_threshold=0.5, color="black"):
         plt.figure()
+        plt.figure(figsize=(self.FIGURE_WIDTH, 8))
 
         df = self.melt()
         cols = [col for col in df.columns if col in self.class_names()]
@@ -549,7 +550,7 @@ def visualize_model_sets(self, min_softmax_threshold=0.5, color="black"):
              show_counts="%d",
              show_percentages="{:.0%}",
              orientation='horizontal',
-             min_subset_size=2)
+             min_subset_size=3)
         plt.savefig(f'{self.output_dir}/upset.png', bbox_inches='tight')
 
     def prediction_sets_df(self, prediction_sets, export_to_dir=None):