根据 WinBugs/JAGS 中的 if - else 条件选择不同的分布

Posted 2023-02-19

【中文标题】根据 WinBugs/JAGS 中的 if - else 条件选择不同的分布

【英文标题】：Choosing Different Distributions based on if - else condition in WinBugs/JAGS

【发布时间】：2013-03-03 02:22:33

【问题描述】：

我正在尝试编写一个 Winbugs/Jags 模型来建模多粒度主题模型（正是这篇论文 -> http://www.ryanmcd.com/papers/mg_lda.pdf）

在这里，我想根据特定值选择不同的分布。 例如：我想做类似的事情

`if ( X[i] > 0.5 )

Z[i] ~ dcat(theta-gl[D[i], 1:K-gl])
W[i] ~ dcat(phi-gl[z[i], 1:V])

else 

Z[i] ~ dcat(theta-loc[D[i], 1:K-loc])
W[i] ~ dcat(phi-loc[z[i], 1:V])

`

这可以在 Winbugs/JAGS 中完成吗？

【问题讨论】：

我不太确定我是否完全理解您的代码。在 WinBUGS 中可以使用 if 语句....mrc-bsu.cam.ac.uk/bugs/faqs/contents.shtml#q15

【参考方案1】：

Winbugs/JAGS 不是过程语言，因此您不能使用这样的结构。使用step 函数。引用手册：

step(e) ...... 1 如果 e >= 0;否则为 0

所以你需要一个技巧来改变条件：

X[i] > 0.5   <=>     
X[i] - 0.5 > 0  <=> 
!(X[i] - 0.5 <= 0) <=>
!(-(X[i] - 0.5) >= 0) <=>
!(step(-(X[i] - 0.5)) == 1) <=>
step(-(X[i] - 0.5)) == 0

然后将其用于索引技巧：

# then branch
Z_branch[i, 1] ~ dcat(theta-gl[D[i], 1:K-gl])
W_branch[i, 1] ~ dcat(phi-gl[z[i], 1:V])

# else branch
Z_branch[i, 2] ~ dcat(theta-loc[D[i], 1:K-loc])
W_branch[i, 2] ~ dcat(phi-loc[z[i], 1:V])

# decision here
if_branch[i] <- 1 + step(-(X[i] - 0.5)) # 1 for "then" branch, 2 for "else" branch
Z[i] ~ Z_branch[i, if_branch[i]]
W[i] ~ W_branch[i, if_branch[i]]

【讨论】：

这里的<=> 是什么？你只是表示它们是等价的吗？

@jbaums 确切地说:-) 这只是一个表示等价的数学符号，而不是实际计算机语言的符号:)