Source code for autofit.non_linear.search.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.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(),
**kwargs
)
def time_series(self, **kwargs):
self._plot_time_series(
samples=self.samples.search_internal.get_chain(),
)