From 529766a985805cbb8d04eff94b66ab81fe3134cd Mon Sep 17 00:00:00 2001 From: Yannick Roehlly Date: Wed, 19 Feb 2014 19:44:51 +0100 Subject: [PATCH] Add chi2 distribution plotting to pdf_analysis --- .../analysis_modules/pdf_analysis/__init__.py | 27 +++++++++++++++++++ 1 file changed, 27 insertions(+) diff --git a/pcigale/analysis_modules/pdf_analysis/__init__.py b/pcigale/analysis_modules/pdf_analysis/__init__.py index dc434a7a..edf76f44 100644 --- a/pcigale/analysis_modules/pdf_analysis/__init__.py +++ b/pcigale/analysis_modules/pdf_analysis/__init__.py @@ -541,5 +541,32 @@ class PdfAnalysis(AnalysisModule): obs_name, var_name)) plt.close(figure) + ################################################################## + # Reduced-chisquares plots # + ################################################################## + if plot_chi2_distribution: + + print("Plotting the reduced chi squares distributions...") + + for obs_index, obs_name in enumerate(obs_table["id"]): + + obs_red_chisquares = reduced_chi_squares[:, obs_index] + + for var_index, var_name in enumerate(analysed_variables): + + values = model_variables[:, obs_index, var_index] + + figure = plt.figure() + ax = figure.add_subplot(111) + ax.plot(values, obs_red_chisquares, "ob") + ax.set_xlabel(var_name) + ax.set_ylabel("reduced chi-square") + figure.suptitle("Reduced chi-square distribution of {} " + "values for {}".format(obs_index, var_name)) + figure.savefig(OUT_DIR + "{}_{}_chisquares.pdf".format( + obs_name, var_name)) + plt.close(figure) + + # AnalysisModule to be returned by get_module Module = PdfAnalysis -- GitLab