pymc3 - PyMC3中的分层建模分类变量交互

Question

我正在尝试使用 PyMC3 来实现具有分类变量及其交互的层次模型。在 R 中，公式将采用以下形式：

y ~ x1 + x2 + x1:x2

但是，在教程https://pymc-devs.github.io/pymc3/GLM-hierarchical/#partial-pooling-hierarchical-regression-aka-the-best-of-both-worlds他们明确说 glm 没有还不能很好地使用分层建模。

那么我将如何添加 x1:x2 项？它会是具有两个分类父项（x1 和 x2）的分类变量吗？

Answer 1

您可以手动将交互项添加到线性模型中。您必须添加 3 个回归系数（β）和一个截距。然后，您可以按如下方式估计您的 y：

y = pm.Normal('regression', 
               mu=intercept + beta_x1 * data_x1 + beta_x2 * data_x2 + beta_interaction * data_x1 * data_x2, 
               sd=sigma, 
               observed=data_y)

参数本身都可以具有超先验来构建分层模型。