plot_param()

Contents

plot_param()#

liesel.goose.plot_param(results, param, param_index=None, chain_indices=None, max_chains=5, max_lags=None, title=None, title_spacing=0.9, style='whitegrid', color_list=None, figure_size=(9, 6), alpha=0.8, legend_position=(1.2, 0.4), save_path=None, show=True)[source]#

Visualizes trace plot, density plot and autocorrelation plot of a single subparameter.

Parameters:

  • results (SamplingResults | dict[str, Any]) – Result object of the sampling process. Must have a method get_posterior_samples() which extracts all samples from the posterior distribution. Alternatively, you can directly supply a dictionary of samples, for example the output of SamplingResults.get_posterior_samples().

  • param (str) – Name of a single model parameter that is contained in the plot. Must coincide with one dictionary key of the Position with the posterior samples.

  • param_index (int | None, default: None) – A single index of the selected model parameter that is contained in the plot. Selects e.g. beta[0] out of a beta parameter vector. Can be specified as an integer or as a sequence containing one integer. If None, the parameter is assumed to have only a single index.

  • chain_indices (int | Sequence[int] | None, default: None) – Indices of chains for each model subparameter that are contained in the plot. Selects e.g. chain 0 and chain 2 out of multiple chains. A single index can be specified as an integer or a sequence containing one integer. If None, all chains are included.

  • max_chains (int | None, default: 5) – Upper bound how many chains are included within each subplot/facet. Avoids overplotting. If None, all chains contained in the results input are plotted. Always starts chain selection from the lowest chain index upwards. For selecting specific chains use the argument chain_indices.

  • max_lags (int | None, default: None) – Maximum number of time lags shown on the x-axis of the autocorrelation plot. If None, the minimum of the chain lengths and 30 is chosen.

  • title (str | None, default: None) – Plot title.

  • title_spacing (float, default: 0.9) – Determines the margin/whitespace between the plot title (set with fig.suptitle()) and the first row of subplots/facets. Passed to the top argument of fig.subplots_adjust().

  • style (str, default: 'whitegrid') – Passed to the style argument of sns.set_theme(). Valid options are darkgrid, whitegrid, dark, white, and ticks.

  • color_list (list[str] | None, default: None) – Determines the chain colors for all three subplots. Custom colors can be passed with a list of color strings. The length of the list must match the number of chains. If None, the default tab10 matplotlib colormap is chosen.

  • figure_size (tuple[int | float, int | float], default: (9, 6)) – Size of the entire plot grid. Passed to the figsize argument of plt.figure(). When changing the figure size consider changing the legend_position as well. Generally, a ratio of 3

  • legend_position (tuple[float, float], default: (1.2, 0.4)) – Determines the color legend position. Coordinates are relative to the upper panel within the plot grid. The first coordinate specifies the horizontal, the second coordinate the vertical position. Might require an adjustment when changing the figure_size values or the number of chains.

  • alpha (float, default: 0.8) – Transparency in trace plot, float in [0,1].

  • save_path (str | None, default: None) – File path where the plot is saved.

Return type:

None

Contents