unmarginalize

pymc_extras.marginal.unmarginalize(model: Model, rvs_to_unmarginalize: str | Sequence[str] | None = None) → Model

Unmarginalize a subset of variables in a PyMC model.

Parameters:

• model (Model) – PyMC model to unmarginalize. Original variables will be cloned.

• rvs_to_unmarginalize (str or sequence of str, optional) – Variables to unmarginalize in the returned model. If None, all variables are unmarginalized.

Returns:

unmarginal_model – Model with the specified variables unmarginalized.

Return type:

Model