Set figure sizes and fonts

Molmed · Nov 23, 2024 · 7e55467 · 7e55467
1 parent 94b66b6
commit 7e55467
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/src/conformist/performance_report.py b/src/conformist/performance_report.py
@@ -5,6 +5,11 @@
 
 
 class PerformanceReport(OutputDir):
+    FIGURE_FONTSIZE = 12
+    FIGURE_WIDTH = 12
+    FIGURE_HEIGHT = 8
+    plt.rcParams.update({'font.size': FIGURE_FONTSIZE})
+
     def __init__(self, base_output_dir):
         self.create_output_dir(base_output_dir)
 
@@ -45,7 +50,8 @@ def _class_report(self,
                       ylabel,
                       color):
         # Reset plt
-        plt.figure()
+        plt.figure(figsize=(self.FIGURE_WIDTH,
+                            self.FIGURE_HEIGHT))
 
         # Sort the dictionary by its values
         mean_sizes = dict(sorted(items_by_class.items(),

diff --git a/src/conformist/validation_trial.py b/src/conformist/validation_trial.py
@@ -8,6 +8,11 @@
 
 
 class ValidationTrial(OutputDir):
+    FIGURE_FONTSIZE = 12
+    FIGURE_WIDTH = 12
+    FIGURE_HEIGHT = 8
+    plt.rcParams.update({'font.size': FIGURE_FONTSIZE})
+
     def __init__(self, runs, class_names=[]):
         self.runs = runs
         self.class_names = class_names
@@ -117,7 +122,8 @@ def run_reports(self, base_output_dir):
         print(f'Reports saved to {self.output_dir}')
 
     def visualize_empirical_fnr(self):
-        plt.figure()
+        plt.figure(figsize=(self.FIGURE_WIDTH,
+                            self.FIGURE_HEIGHT))
 
         # Generate a pastel palette
         color_palette = sns.color_palette("deep")