plot_hdi#

pymc_marketing.plot.plot_hdi(curve, non_grid_names, hdi_prob=None, hdi_kwargs=None, subplot_kwargs=None, plot_kwargs=None, axes=None, same_axes=False, colors=None, legend=False, sel_to_string=None)[source]#

Plot hdi of the curve across coords.

Parameters:
curvexr.DataArray

Curve to plot

non_grid_namesstr | set[str]

The names to exclude from the grid. chain and draw are excluded automatically

hdi_probfloat | list[float], optional

HDI probabilities. Defaults to None which uses arviz default for stats.ci_prob which is 94%

hdi_kwargsdict, optional

Kwargs for the arviz.hdi function

same_axesbool

All of the plots in the same axis

colorsIterable[str], optional

Colors for the plots

legendbool, optional

If to include a legend. Defaults to True if same_axes

axesnpt.NDArray[plt.Axes], optional

Axes to plot on

subplot_kwargsdict, optional

Additional kwargs to while creating the fig and axes

plot_kwargsdict, optional

Kwargs for the plot function

Returns:
tuple[plt.Figure, npt.NDArray[plt.Axes]]

Figure and the axes