pymc3 - 如何向 PyMC3 模型添加约束？

Question

如果我们考虑 PyMC3 的以下线性回归示例:

http://docs.pymc.io/notebooks/getting_started.html#A-Motivating-Example:-Linear-Regression

我们如何包含诸如 a + b1 + b2 = 1 或 a^2 + b1^2 = 25 之类的约束？

我知道我们可以使用 Bound 来为变量创建边界，但我不确定如何添加更复杂的约束。

感谢您的帮助!

Answer 1

一般的解决方案是使用电位。

const = pm.Potential('const', pm.math.switch(pm.math.eq(a**2 + b1**2, 25),
                                             0,
                                             -np.inf))

潜力是您可以添加到模型可能性中的任意因素。在此示例中，如果参数满足您的约束，则不添加任何内容，否则添加 -inf。

您也可以提问 here 以供将来引用