Где находится scipy.stats.dirichlet_multinomial.rvs? ⇐ Python

[Anonymous]

Я хотел получить образцы из полиномиального распределения Дирихле, используя SciPy. К сожалению, похоже, что scipy.stats.dirichlet_multinomial не определяет метод rvs, который другие дистрибутивы используют для генерации случайных выборок.
Я думаю, это будет эквивалентно для одного примера следующее:

Код: Выделить всё

import scipy.stats as sps

def dirichlet_multinomial_sample(alpha, n, **kwargs):
kwargs['size'] = 1 # force size to 1 for simplicity
p = sps.dirichlet.rvs( alpha=alpha, **kwargs )
return sps.multinomial.rvs( n=n, p=p.ravel(), **kwargs )

Несколько выборок (т. е. размер > 1) можно нарисовать аналогичным образом, приложив немного больше усилий, чтобы сделать это более эффективным. Кажется, это достаточно легко реализовать. У меня два вопроса:

• Правильна ли приведенная выше реализация?
• Если да, то как я могу предложить это улучшение? разработчикам SciPy?

Подробнее здесь: https://stackoverflow.com/questions/792 ... nomial-rvs