plot_curve

pymc_marketing.mmm.plot.plot_curve(curve, non_grid_names, subplot_kwargs=None, sample_kwargs=None, hdi_kwargs=None)

Plot HDI with samples of the curve across coords.

Parameters:

curvexr.DataArray

Curve to plot

non_grid_namesset[str]

The names to exclude from the grid. HDI and samples both have defaults of hdi and chain, draw, respectively

subplot_kwargsdict, optional

Addtional kwargs to while creating the fig and axes

sample_kwargsdict, optional

Kwargs for the plot_curve() function

hdi_kwargsdict, optional

Kwargs for the plot_hdi() function

Returns:

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

Figure and the axes