python - Python 包：用于 Dirichlet 分发的 MLE

Question

我想知道是否有人知道一个 python 包，它实现了 MLE 来估计 Dirichlet 分布的参数。

Answer 1

Eric Suh在这里有一个包裹。

$ pip install git+https://github.com/ericsuh/dirichlet.git

然后：

import numpy
import dirichlet
a0 = numpy.array([100, 299, 100])
D0 = numpy.random.dirichlet(a0, 1000)
dirichlet.mle(D0)