pymc.Dirichlet

class pymc.Dirichlet(name, *args, rng=None, dims=None, initval=None, observed=None, total_size=None, transform=UNSET, **kwargs)

狄利克雷对数似然。

\[f(\mathbf{x}|\mathbf{a}) = \frac{\Gamma(\sum_{i=1}^k a_i)}{\prod_{i=1}^k \Gamma(a_i)} \prod_{i=1}^k x_i^{a_i - 1}\]

支持	\(x_i \in (0, 1)\) 对于 \(i \in \{1, \ldots, K\}\) 使得 \(\sum x_i = 1\)
均值	\(\dfrac{a_i}{\sum a_i}\)
方差	\(\dfrac{a_i - \sum a_0}{a_0^2 (a_0 + 1)}\) 其中 \(a_0 = \sum a_i\)

参数:

a : 类张量 的 floattensor_like of float

浓度参数（a > 0）。类别的数量由最后一个轴的长度给出。

方法

Dirichlet.dist(a, **kwargs)

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