通过变量枚举(将PyMC移植到PyMC3)

Question

我从PyMC3开始，将this code从PyMC转换为PyMC3。

我不知道如何翻译这段话：

v = pymc.Beta('v', alpha=1, beta=alpha, size=N_dp)
@pymc.deterministic
def p(v=v):
    """ Calculate Dirichlet probabilities """

    # Probabilities from betas
    # this line creates the error:
    value = [u*np.prod(1-v[:i]) for i,u in enumerate(v)]
    # Enforce sum to unity constraint
    value[-1] = 1-sum(value[:-1])

    return value

z = pymc.Categorical('z', p, size=len(set(counties)))

有没有人可以教我怎么翻译或者把我链接到文档中的相关部分？谢谢。

Answer 1

Dirichlet发行版实际上是built into pymc3，因此整个代码块可以替换为：

with pm.Model():
    ...
    v = pm.Beta('v', alpha=1, beta=alpha, shape=N_dp)
    p = pm.Dirichlet('p', a=v, shape=N_dp)
    ...
    trace = pm.sample(20000)