pymc.DirichletMultinomial

class pymc.DirichletMultinomial(name, *args, **kwargs)

Dirichlet 多项式对数似然。

多项分布的狄利克雷混合分布，具有边缘化的概率质量函数。

\[f(x \mid n, a) = \frac{\Gamma(n + 1)\Gamma(\sum a_k)} {\Gamma(n + \sum a_k)} \prod_{k=1}^K \frac{\Gamma(x_k + a_k)} {\Gamma(x_k + 1)\Gamma(a_k)}\]

支持	\(x \in \{0, 1, \ldots, n\}\) 使得 \(\sum x_i = n\)
均值	\(n \frac{a_i}{\sum{a_k}}\)

参数:

n : 类似张量 的 intpython:int 的 tensor_like

每个重复中的总计数（n > 0）。

a : 类张量 的 floattensor_like of float

Dirichlet 集中参数 (a > 0)。类别的数量由最后一个轴的长度给出。

方法

DirichletMultinomial.dist(n, a, *args, **kwargs)

创建一个与 cls 分布相对应的张量变量。