From 3f358a4f3f8b3e2f353177a9bad67737bbdc983a Mon Sep 17 00:00:00 2001 From: Allan Denis Date: Mon, 25 Nov 2024 16:00:08 +0100 Subject: [PATCH] AD : small update on plot_corner : Adding a figsize --- ForMoSA/plotting/plotting_class.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ForMoSA/plotting/plotting_class.py b/ForMoSA/plotting/plotting_class.py index 855897e..cbffa0b 100644 --- a/ForMoSA/plotting/plotting_class.py +++ b/ForMoSA/plotting/plotting_class.py @@ -336,7 +336,7 @@ def _get_posteriors(self): self.posteriors_names = tot_list_param_title - def plot_corner(self, levels_sig=[0.997, 0.95, 0.68], bins=100, quantiles=(0.16, 0.5, 0.84), burn_in=0): + def plot_corner(self, levels_sig=[0.997, 0.95, 0.68], bins=100, quantiles=(0.16, 0.5, 0.84), burn_in=0, figsize=(15,15)): ''' Function to display the corner plot @@ -353,6 +353,7 @@ def plot_corner(self, levels_sig=[0.997, 0.95, 0.68], bins=100, quantiles=(0.16, # make sure posteriors are loaded self._get_posteriors() + fig = plt.figure(figsize=figsize) fig = corner.corner(self.posterior_to_plot[burn_in:], weights=self.weights[burn_in:], labels=self.posteriors_names, @@ -371,6 +372,7 @@ def plot_corner(self, levels_sig=[0.997, 0.95, 0.68], bins=100, quantiles=(0.16, title_kwargs=dict(fontsize=14), contour_kwargs=dict(colors=self.color_out, linewidths=0.7), pcolor_kwargs=dict(color='red'), + fig=fig, label_kwargs=dict(fontsize=14))