dynode.utils.vis_utils.plot_mcmc_chains

dynode.utils.vis_utils.plot_mcmc_chains(samples_in: dict[str, ndarray | list], matplotlib_style: list[str] | str = ['seaborn-v0_8-colorblind']) → Figure

Plot MCMC chains for each sampled parameter in a grid of subplots.

Parameters

samples_indict[str, np.ndarray | list]

Dictionary containing posterior samples (shape: num_chains, num_samples). Parameters generated with numpyro.plate are flattened and displayed as separate parameters with _i suffix for each i in num_plates.

matplotlib_stylelist[str] | str, optional

Matplotlib style to use for plotting; default is [“seaborn-v0_8-colorblind”].

Returns

plt.Figure

Matplotlib figure containing the plots.