pymc3 - Pymc3 のカテゴリー混合モデル

Question

私は Pymc3 が初めてで、https://en.wikipedia.org/wiki/Mixture_model#Categorical_mixture_modelに示されているカテゴリ混合モデルを作成しようとしています。「x」変数を接続するのに苦労しています。z変数を決定論的にする必要があるためだと思いますが、「x」が割り当てられている行でエラーメッセージが表示されます：「ValueError：3つの入力が必要ですが、2を取得しました。」. p関数は2つの入力しか受け付けないように見えるので、行き詰まっています。いろいろ試してみましたが、まだこれを機能させることができませんでした。

import numpy as np
from pymc3 import *
import theano.tensor as t

K = 3 #NUMBER OF TOPICS
V = 20 #NUMBER OF WORDS
N = 15 #NUMBER OF DOCUMENTS

#GENERAETE RANDOM CATEGORICAL MIXTURES
data = np.ones([N,V])

@theano.compile.ops.as_op(itypes=[t.lscalar, t.dscalar, t.dscalar],otypes=[t.dvector])
def p(z=z, phi=phi):
    return [phi[z[i,j]] for i in range(D) for j in range(W)]

model = Model()
with model:

    alpha = np.ones(V)
    beta = np.ones(K)

    theta = [Dirichlet('theta_%i' % i, alpha, shape=V) for i in range(K)]
    phi = Dirichlet('phi', beta, shape=K)

    z = [Categorical('z_%i' % i, p = phi, shape=V) for i in range(N)]
    x = [Categorical('x_%i_%i' % (i,j), p=p(z[i][j],phi), observed=data[i,j]) for i in range(N) for j in range(V)]
    #x = [Categorical('x_%i_%i' % (i,j), p=theta[z[i][j]], observed=data[i,j]) for i in range(N) for j in range(V)]

    print "Created model.  Now begin sampling"
    step = Slice()
    trace = sample(n, step)

    trace.get_values('phi')