numpy.random.dirichlet

random.dirichlet(alpha, size=None)

Выборка образцов из распределения Дирихле.

Выборка size образцов размерности k из распределения Дирихле. Случайная величина, распределённая по Дирихле, представляет собой многомерное обобщение бета-распределения. Распределение Дирихле является сопряжённым априорным распределением для многономиального распределения в байесовском выводе.

Примечание

Новый код должен использовать метод dirichlet экземпляра Generator; см. Быстрый старт.

Параметры:

alphaпоследовательность чисел с плавающей точкой, длина k

Параметр распределения (длина k для образца длины k).

sizeцелое число или кортеж целых чисел, необязательно

Форма выходных данных. Если заданная форма, например, (m, n), то m * n * k образцов будут отобраны. По умолчанию None, в этом случае возвращается вектор длины k.

Возвращает:

samplesndarray,

Выбранные образцы, с формой (size, k).

Возбуждает:

ValueError

Если любое значение в alpha меньше или равно нулю

См. также

random.Generator.dirichlet

который следует использовать для нового кода.

Примечания

Распределение Дирихле — это распределение над векторами \(x\), которые удовлетворяют условиям \(x_i>0\) и \(\sum_{i=1}^k x_i = 1\).

Функция плотности вероятности \(p\) случайного вектора \(X\), распределённого по Дирихле, пропорциональна

\[p(x) \propto \prod_{i=1}^{k}{x^{\alpha_i-1}_i},\]

где \(\alpha\) — вектор, содержащий положительные параметры концентрации.

Этот метод использует следующее свойство для вычисления: пусть \(Y\) — случайный вектор, компоненты которого следуют стандартному гамма-распределению, тогда \(X = \frac{1}{\sum_{i=1}^k{Y_i}} Y\) распределено по Дирихле.

Ссылки

[1]

David McKay, “Теория информации, вывод и алгоритмы обучения,” глава 23, https://www.inference.org.uk/mackay/itila/

[2]

Википедия, “Распределение Дирихле”, https://en.wikipedia.org/wiki/Dirichlet_distribution

Примеры

Рассмотрим пример, приведённый в Википедии, это распределение может быть использовано, если требуется разбить строки (каждая начальной длины 1.0) на K частей с различными длинами, где каждая часть имела бы в среднем заданную среднюю длину, но позволяя некоторую вариацию в относительных размерах частей.

>>> s = np.random.dirichlet((10, 5, 3), 20).transpose()

>>> import matplotlib.pyplot as plt
>>> plt.barh(range(20), s[0])
>>> plt.barh(range(20), s[1], left=s[0], color='g')
>>> plt.barh(range(20), s[2], left=s[0]+s[1], color='r')
>>> plt.title("Lengths of Strings")