11

マルチグレイントピックモデルをモデル化するための 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 で実行できますか?

4

1 に答える 1

10

Winbugs/JAGS は手続き型言語ではないため、そのような構造を使用することはできません。関数を使用stepします。マニュアルからの引用：

step(e) ...... e >= 0 の場合は 1。それ以外の場合は 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]]
于 2013-12-11T21:13:25.340 に答える