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:

  • *params (Any) – One or more ModelTerm, Sum, Product, or compositions.

  • ds (Dataset) – Dataset containing the data variables.

Return type:

dict[str, Any]

Returns:

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:
    ...