Source code for autofit.non_linear.mcmc.zeus.plotter

from autofit.plot.samples_plotters import MCMCPlotter

import zeus

[docs]class ZeusPlotter(MCMCPlotter): def corner(self, **kwargs): try: zeus.cornerplot( samples=self.samples.results_internal.get_chain(flat=True), weight_list=self.samples.weight_list, labels=self.model.parameter_labels_with_superscripts_latex, **kwargs ) except TypeError: pass self.output.to_figure(structure=None, auto_filename="corner") def trajectories(self, **kwargs): self._plot_trajectories( samples=self.samples.results_internal.get_chain(), log_posterior_list=self.samples.results_internal.get_log_prob(), **kwargs ) def likelihood_series(self, **kwargs): self._plot_likelihood_series( log_posterior_list = self.samples.results_internal.get_log_prob() ) def time_series(self, **kwargs): self._plot_time_series( samples=self.samples.results_internal.get_chain() )