marginalize

pymc_extras.marginalize(model: Model, rvs_to_marginalize: TensorVariable | Sequence[TensorVariable] | str | Sequence[str]) → MarginalModel

边缘化 PyMC 模型中的变量子集。

这将从现有的 Model 创建一个 MarginalModel 类，并将指定的变量边缘化。

有关更多信息，请参阅 MarginalModel 的文档。

参数:

• model (Model) – 要边缘化的 PyMC 模型。原始变量将被克隆。

• rvs_to_marginalize (Sequence[TensorVariable]) – 要在返回模型中边缘化的变量。

返回:

marginal_model – 具有指定变量边缘化的边缘模型。

返回类型:

MarginalModel