collect_coords#

pymc_marketing.terms.collect_coords(*params, ds)[source]#

Collect coordinates from multiple parameter expressions.

Convenience wrapper for collecting coordinates from several independent term trees (e.g., a mu expression and a sigma expression).

Parameters:
*paramsAny

One or more ModelTerm, Sum, Product, or compositions.

dsxr.Dataset

Dataset containing the data variables.

Returns:
dict[str, Any]

Combined coordinates dict for pm.Model.

Examples

coords = collect_coords(mu, sigma, ds=ds)
coords["obs"] = ds.coords["obs"].values.tolist()

with pm.Model(coords=coords) as model:
    ...