基于“`rstanarm`”包中的`stan_glm()`分组变量的后验预测？

Question

我想知道如何从stan_glm()包中的rstanarm包中获得基于分组变量的后验预测？

例如，如果数据中有一个名为(0, 1)的二进制"vs"编码分组变量(基R数据：mtcars)，那么如何获得对何时vs == 0和何时vs == 1的预测？

这是我的R码：

library(rstanarm)
fit <- stan_glm(mpg ~., data = mtcars)

posterior_predict(fit, newdata = WHAT SHOULD BE HERE?)

Answer 1

为了探讨例如vs对结果的影响(在您的例子中是mpg)，您可以分别在vs == 0和vs == 1的子集上使用posterior_predict：

posterior_predict(fit, newdata = subset(mtcars[1:10, ], vs == 0));

和

posterior_predict(fit, newdata = subset(mtcars[1:10, ], vs == 1));

更多细节在?rstanarm::posterior_predict中给出。