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

import numpy as np
import corner

from autofit.plot.samples_plotters import MCMCPlotter

[docs]class EmceePlotter(MCMCPlotter): def corner(self, **kwargs): corner.corner( data=np.asarray(self.samples.parameter_lists), weight_list=self.samples.weight_list, labels=self.model.parameter_labels_with_superscripts_latex, **kwargs ) self.output.to_figure(structure=None, auto_filename="corner") self.close() 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(), **kwargs ) def time_series(self, **kwargs): self._plot_time_series( samples=self.samples.results_internal.get_chain(), )