Source code for autofit.non_linear.search.mcmc.zeus.plotter
from autofit.plot.samples_plotters import MCMCPlotter
[docs]class ZeusPlotter(MCMCPlotter):
def corner(self, **kwargs):
import zeus
try:
zeus.cornerplot(
samples=self.samples.search_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.search_internal.get_chain(),
log_posterior_list=self.samples.search_internal.get_log_prob(),
**kwargs
)
def likelihood_series(self, **kwargs):
self._plot_likelihood_series(
log_posterior_list = self.samples.search_internal.get_log_prob()
)
def time_series(self, **kwargs):
self._plot_time_series(
samples=self.samples.search_internal.get_chain()
)