tf.compat.v1.distributions.Distribution
Базовый класс вероятностного распределения общего назначения.
tf.compat.v1.distributions.Distribution(
dtype, reparameterization_type, validate_args, allow_nan_stats, parameters=None,
graph_parents=None, name=None
)
Distribution — это базовый класс для построения и организации свойств (например, среднего значения, дисперсии) случайных величин (например, Бернулли, Гаусса).
Наследование
Ожидается, что подклассы будут реализовывать функцию с ведущим подчеркиванием с тем же именем. Подпись аргументов должна быть идентичной, за исключением пропуска name="...". Например, для включения log_prob(value, name="log_prob") подкласс должен реализовать _log_prob(value).
Подклассы могут дополнять общедоступные строки документации, предоставляя строки документации для своих специализаций методов. Например:
@util.AppendDocstring("Some other details.")
def _log_prob(self, value):
...
добавит строку "Дополнительные сведения." к строке документации функции log_prob. Это реализовано как простой декоратор, чтобы избежать сообщений линтера Python о недостающих разделах Args/Returns/Raises в частичных строках документации.
Вещание, пакетные операции и формы
Все распределения поддерживают пакеты независимых распределений того же типа. Форма пакета определяется путем вещания параметров.
Форма аргументов для __init__, cdf, log_cdf, prob, и log_prob отражает это вещание, как и возвращаемое значение sample и sample_n.
sample_n_shape = [n] + batch_shape + event_shape, где sample_n_shape — форма Tensor возвращаемого значением sample_n, n — количество выборок, batch_shape определяет, сколько независимых распределений существует, а event_shape определяет форму выборок из каждого из этих независимых распределений. Выборки независимы вдоль измерений batch_shape, но необязательно независимы вдоль измерений event_shape (в зависимости от особенностей базового распределения).
В качестве примера использования распределения Uniform:
minval = 3.0
maxval = [[4.0, 6.0],
[10.0, 12.0]]
# Broadcasting:
# This instance represents 4 Uniform distributions. Each has a lower bound at
# 3.0 as the `minval` parameter was broadcasted to match `maxval`'s shape.
u = Uniform(minval, maxval)
# `event_shape` is `TensorShape([])`.
event_shape = u.event_shape
# `event_shape_t` is a `Tensor` which will evaluate to [].
event_shape_t = u.event_shape_tensor()
# Sampling returns a sample per distribution. `samples` has shape
# [5, 2, 2], which is [n] + batch_shape + event_shape, where n=5,
# batch_shape=[2, 2], and event_shape=[].
samples = u.sample_n(5)
# The broadcasting holds across methods. Here we use `cdf` as an example. The
# same holds for `log_cdf` and the likelihood functions.
# `cum_prob` has shape [2, 2] as the `value` argument was broadcasted to the
# shape of the `Uniform` instance.
cum_prob_broadcast = u.cdf(4.0)
# `cum_prob`'s shape is [2, 2], one per distribution. No broadcasting
# occurred.
cum_prob_per_dist = u.cdf([[4.0, 5.0],
[6.0, 7.0]])
# INVALID as the `value` argument is not broadcastable to the distribution's
# shape.
cum_prob_invalid = u.cdf([4.0, 5.0, 6.0])
Формы
С распределениями TensorFlow связано три важных понятия:
- Форма события описывает форму одного вывода из распределения; она может зависеть от измерений. Для скалярных распределений форма события равна
[]. Для 5-мерного MultivariateNormal форма события равна[5]. - Форма пакета описывает независимые, не одинаково распределённые выводы, т.е. «коллекцию» или «группу» распределений.
- Форма выборки описывает независимые, одинаково распределённые выводы из семейства распределений.
Форма события и форма пакета являются свойствами объекта Distribution, тогда как форма выборки связана с конкретным вызовом sample или log_prob.
Подробные примеры использования форм распределений TensorFlow см. в этом руководстве
Значения параметров, приводящие к неопределенным статистическим данным или распределениям
Некоторые распределения не имеют хорошо определённых статистических данных для всех значений параметров инициализации. Например, распределение бета параметризуется положительными действительными числами concentration1 и concentration0, и не имеет хорошо определённой моды, если concentration1 < 1 или concentration0 < 1.
Пользователь может выбрать поднятие исключения или возвращение NaN.
a = tf.exp(tf.matmul(logits, weights_a)) b = tf.exp(tf.matmul(logits, weights_b)) # Will raise exception if ANY batch member has a < 1 or b < 1. dist = distributions.beta(a, b, allow_nan_stats=False) mode = dist.mode().eval() # Will return NaN for batch members with either a < 1 or b < 1. dist = distributions.beta(a, b, allow_nan_stats=True) # Default behavior mode = dist.mode().eval()
Во всех случаях при передаче неправильных параметров поднимается исключение, например,
# Will raise an exception if any Op is run. negative_a = -1.0 * a # beta distribution by definition has a > 0. dist = distributions.beta(negative_a, b, allow_nan_stats=True) dist.mean().eval()
| Аргументы | |
|---|---|
dtype | Тип событий выборок. None подразумевает отсутствие проверки типа. |
reparameterization_type | Экземпляр ReparameterizationType. Если distributions.FULLY_REPARAMETERIZED, это Distribution можно перепараметризовать в терминах некоторого стандартного распределения с помощью функции, чья якобианна постоянна для области определения стандартного распределения. Если distributions.NOT_REPARAMETERIZED, то такая перепараметризация недоступна. |
validate_args | Python bool, по умолчанию False. При проверке параметров распределения True на валидность, несмотря на возможное ухудшение производительности во время выполнения. При False недопустимые входные данные могут молча привести к некорректным результатам. |
allow_nan_stats | Python bool, по умолчанию True. При True, для статистических данных (например, среднего, моды, дисперсии) используется значение "NaN", чтобы указать, что результат не определён. При False, если один или несколько элементов пакета статистики не определены, поднимается исключение. |
parameters | Python dict параметров, используемых для создания этого Distribution. |
graph_parents | Python list графовых предварительных условий этого Distribution. |
name | Python str имени, добавляемого к операциям, созданным этим классом. По умолчанию: имя подкласса. |
| Исключения | |
|---|---|
ValueError | если любой член graph_parents None или не является Tensor. |
| Атрибуты | |
|---|---|
allow_nan_stats | Python bool, описывающий поведение при неопределённом значении статистики. Статистики возвращают +/- бесконечность, когда это имеет смысл. Например, дисперсия распределения Коши равна бесконечности. Однако иногда статистика не определена, например, если плотность вероятности распределения не достигает максимума в пределах области определения распределения, мода не определена. Если среднее значение не определено, то по определению дисперсия не определена. Например, среднее значение распределения Стьюдента для df = 1 не определено (нет ясного способа сказать, что это либо + или - бесконечность), поэтому дисперсия = E[(X - среднее)**2] также не определена. |
batch_shape | Форма одной выборки из одного индекса события как TensorShape. Может быть частично определена или неизвестна. Размеры пакета — индексы в независимых, не идентичных параметризациях этого распределения. |
dtype | DType Tensor обрабатываемых этим Distribution. |
event_shape | Форма одной выборки из одного пакета как TensorShape. Может быть частично определена или неизвестна. |
name | Имя, добавляемое ко всем операциям, созданным этим Distribution. |
parameters | Словарь параметров, используемых для создания этого Distribution. |
reparameterization_type | Описывает, как выборки из распределения перепараметризуются. В настоящее время это один из статических экземпляров |
validate_args | Python bool, указывающий на то, что возможно дорогостоящие проверки включены. |
Методы
batch_shape_tensor
batch_shape_tensor(
name='batch_shape_tensor'
)
Форма одной выборки из одного индекса события как одномерный Tensor.
Размеры пакета — это индексы в независимых, не идентичных параметризациях этого распределения.
| Аргументы | |
|---|---|
name | имя, присваиваемое операции |
| Возвращаемые значения | |
|---|---|
batch_shape | Tensor. |
cdf
cdf(
value, name='cdf'
)
Функция кумулятивного распределения.
Для случайной величины X, функция кумулятивного распределения cdf:
cdf(x) := P[X <= x]
| Аргументы | |
|---|---|
value | float или double Tensor. |
name | Python str, добавляемое к именам операций, созданных этой функцией. |
| Возвращаемые значения | |
|---|---|
cdf | Tensor формы sample_shape(x) + self.batch_shape со значениями типа self.dtype. |
copy
copy(
**override_parameters_kwargs
)
Создаёт глубокую копию распределения.
Примечание: скопированное распределение может по-прежнему зависеть от исходных аргументов инициализации.
| Аргументы | |
|---|---|
**override_parameters_kwargs | Словарь строк/значений аргументов инициализации для переопределения новыми значениями. |
| Возвращаемое значение | |
|---|---|
distribution | Новый экземпляр type(self), инициализированный из объединения self.parameters и override_parameters_kwargs, т.е. dict(self.parameters, **override_parameters_kwargs). |
covariance
covariance(
name='covariance'
)
Ковариация.
Ковариация (возможно) определена только для распределений с нескалярным событием.
Например, для векторного распределения длиной k, она рассчитывается как,
Cov[i, j] = Covariance(X_i, X_j) = E[(X_i - E[X_i]) (X_j - E[X_j])]
где Cov — (векторная) матрица k x k, 0 <= (i, j) < k, а E обозначает математическое ожидание.
В качестве альтернативы, для не-векторных многомерных распределений (например, матричнозначных, Вишарта), Covariance должна возвращать (векторную) матрицу в рамках некоторой векторизации событий, т.е.,
Cov[i, j] = Covariance(Vec(X)_i, Vec(X)_j) = [as above]
где Cov — (векторная) матрица k' x k', 0 <= (i, j) < k' = reduce_prod(event_shape), а Vec — некоторая функция, сопоставляющая индексы размерностей события этого распределения с индексами вектора длиной k'.
| Аргументы | |
|---|---|
name | Префикс Python str для имён операций, созданных этой функцией. |
| Возвращаемое значение | |
|---|---|
covariance | Вещественное число Tensor с формой [B1, ..., Bn, k', k'], где первые n измерения — координаты пакета, а k' = reduce_prod(self.event_shape). |
cross_entropy
cross_entropy(
other, name='cross_entropy'
)
Вычисляет перекрёстную энтропию (Шеннона).
Обозначим это распределение (self) как P и распределение other как Q. Предполагая, что P, Q абсолютно непрерывны друг относительно друга и допускают плотности p(x) dr(x) и q(x) dr(x), (энтропия Шеннона) перекрёстная энтропия определяется как:
H[P, Q] = E_p[-log q(X)] = -int_F p(x) log q(x) dr(x)
где F обозначает область определения случайной величины X ~ P.
| Аргументы | |
|---|---|
other | Экземпляр tfp.distributions.Distribution. |
name | Префикс Python str для имён операций, созданных этой функцией. |
| Возвращаемое значение | |
|---|---|
cross_entropy | Вещественное число self.dtype Tensor с формой [B1, ..., Bn], представляющее n различные вычисления перекрёстной энтропии (Шеннона). |
entropy
entropy(
name='entropy'
)
Энтропия Шеннона в натах.
event_shape_tensor
event_shape_tensor(
name='event_shape_tensor'
)
Форма одного образца из одного пакета как 1-мерный int32 Tensor.
| Аргументы | |
|---|---|
name | Имя операции |
| Возвращаемое значение | |
|---|---|
event_shape | Tensor. |
is_scalar_batch
is_scalar_batch(
name='is_scalar_batch'
)
Указывает, что batch_shape == [].
| Аргументы | |
|---|---|
name | Префикс Python str для имён операций, созданных этой функцией. |
| Возвращаемое значение | |
|---|---|
is_scalar_batch | Скаляр bool Tensor. |
is_scalar_event
is_scalar_event(
name='is_scalar_event'
)
Указывает, что event_shape == [].
| Аргументы | |
|---|---|
name | Префикс Python str для имён операций, созданных этой функцией. |
| Возвращаемое значение | |
|---|---|
is_scalar_event | Скаляр bool Tensor. |
kl_divergence
kl_divergence(
other, name='kl_divergence'
)
Вычисляет расстояние Кульбака—Лайблера.
Обозначим это распределение (self) как p и распределение other как q. Предполагая, что p, q абсолютно непрерывны относительно меры отсчета r, расстояние Кульбака—Лайблера определяется как:
KL[p, q] = E_p[log(p(X)/q(X))]
= -int_F p(x) log q(x) dr(x) + int_F p(x) log p(x) dr(x)
= H[p, q] - H[p]
где F обозначает область определения случайной величины X ~ p, H[., .] обозначает перекрёстную энтропию (Шеннона), а H[.] обозначает энтропию (Шеннона).
| Аргументы | |
|---|---|
other | Экземпляр tfp.distributions.Distribution. |
name | Префикс Python str для имён операций, созданных этой функцией. |
| Возвращаемое значение | |
|---|---|
kl_divergence | Вещественное число self.dtype Tensor с формой [B1, ..., Bn], представляющее n различные вычисления расстояния Кульбака—Лайблера. |
log_cdf
log_cdf(
value, name='log_cdf'
)
Логарифм кумулятивной функции распределения.
Для случайной величины X, кумулятивная функция распределения cdf равна:
log_cdf(x) := Log[ P[X <= x] ]
Часто для log_cdf(x) используется численное приближение, которое даёт более точный результат, чем просто взятие логарифма от cdf, когда x << -1.
| Аргументы | |
|---|---|
value | float или double Tensor. |
name | Префикс Python str для имён операций, созданных этой функцией. |
| Возвращаемое значение | |
|---|---|
logcdf | Tensor формы sample_shape(x) + self.batch_shape со значениями типа self.dtype. |
log_prob
log_prob(
value, name='log_prob'
)
Логарифм плотности/массы вероятности.
| Аргументы | |
|---|---|
value | float или double Tensor. |
name | Префикс Python str для имён операций, созданных этой функцией. |
| Возвращаемое значение | |
|---|---|
log_prob | Tensor формы sample_shape(x) + self.batch_shape со значениями типа self.dtype. |
log_survival_function
log_survival_function(
value, name='log_survival_function'
)
Логарифм функции выживания.
Для случайной величины X, функция выживания определяется как:
log_survival_function(x) = Log[ P[X > x] ]
= Log[ 1 - P[X <= x] ]
= Log[ 1 - cdf(x) ]
Обычно для логарифма функции выживания используются различные численные приближения, которые более точны, чем 1 - cdf(x) при x >> 1.
| Аргументы | |
|---|---|
value | float или double Tensor. |
name | Префикс Python str для имён операций, созданных этой функцией. |
| Возвращаемое значение | |
|---|---|
Tensor формы sample_shape(x) + self.batch_shape со значениями типа self.dtype. |
mean
mean(
name='mean'
)
Среднее значение.
mode
mode(
name='mode'
)
Мода.
param_shapes
@classmethod
param_shapes(
sample_shape, name='DistributionParamShapes'
)
Формы параметров, заданных желаемой формой вызова sample().
Это метод класса, который описывает, какие аргументы ключ/значение необходимы для создания экземпляра данного Distribution, чтобы получить определённую форму для вызова sample() этим экземпляром.
Подклассы должны переопределить метод класса _param_shapes.
| Аргументы | |
|---|---|
sample_shape | Tensor или список/кортеж Python. Желаемая форма вызова sample(). |
name | Имя для добавления к операциям. |
| Возвращаемое значение | |
|---|---|
dict от имени параметра до форм Tensor. |
param_static_shapes
@classmethod
param_static_shapes(
sample_shape
)
param_shapes со статическими (т.е. TensorShape) формами.
Это метод класса, который описывает, какие аргументы ключ/значение необходимы для создания экземпляра данного Distribution, чтобы получить определённую форму для вызова sample() этим экземпляром. Предполагается, что форма образца известна статически.
Подклассы должны переопределить метод класса _param_shapes, чтобы возвращать тензоры с постоянными значениями при подаче постоянных значений.
| Аргументы | |
|---|---|
sample_shape | TensorShape или список/кортеж Python. Желаемая форма вызова sample(). |
| Возвращаемое значение | |
|---|---|
dict от имени параметра до TensorShape. |
| Исключения | |
|---|---|
ValueError | если sample_shape является TensorShape и не полностью определён. |
prob
prob(
value, name='prob'
)
Функция плотности/массы вероятности.
| Аргументы | |
|---|---|
value | float или double Tensor. |
name | Python-строка, добавляемая к именам созданных этой функцией операций. |
| Возвращаемое значение | |
|---|---|
prob | Tensor формы sample_shape(x) + self.batch_shape со значениями типа self.dtype. |
quantile
quantile(
value, name='quantile'
)
Функция квантиля. Также известна как «обратная функция распределения» или «функция процентиля».
Для случайной величины X и p in [0, 1], квантиль — это:
quantile(p) := x such that P[X <= x] == p
| Аргументы | |
|---|---|
value | float или double Tensor. |
name | Python-строка, добавляемая к именам созданных этой функцией операций. |
| Возвращаемое значение | |
|---|---|
quantile | Tensor формы sample_shape(x) + self.batch_shape со значениями типа self.dtype. |
sample
sample(
sample_shape=(), seed=None, name='sample'
)
Генерирует образцы указанной формы.
Обратите внимание, что вызов sample() без аргументов сгенерирует один образец.
| Аргументы | |
|---|---|
sample_shape | 0-мерный или 1-мерный int32 Tensor. Форма сгенерированных образцов. |
seed | Целое число Python для генератора случайных чисел (RNG) |
name | Имя для операции. |
| Возвращаемое значение | |
|---|---|
samples | Tensor с добавленными измерениями sample_shape. |
stddev
stddev(
name='stddev'
)
Среднеквадратическое отклонение.
Среднеквадратическое отклонение определяется как,
stddev = E[(X - E[X])**2]**0.5
где X — случайная величина, связанная с этим распределением, E обозначает математическое ожидание, и stddev.shape = batch_shape + event_shape.
| Аргументы | |
|---|---|
name | Python-строка, добавляемая к именам созданных этой функцией операций. |
| Возвращаемое значение | |
|---|---|
stddev | Вещественное число Tensor с формой, идентичной batch_shape + event_shape, т.е. такой же формой, как у self.mean(). |
survival_function
survival_function(
value, name='survival_function'
)
Функция выживания.
Для случайной величины X, функция выживания определяется как:
survival_function(x) = P[X > x]
= 1 - P[X <= x]
= 1 - cdf(x).
| Аргументы | |
|---|---|
value | float или double Tensor. |
name | Python-строка, добавляемая к именам созданных этой функцией операций. |
| Возвращаемое значение | |
|---|---|
Tensor формы sample_shape(x) + self.batch_shape со значениями типа self.dtype. |
variance
variance(
name='variance'
)
Дисперсия.
Дисперсия определяется как,
Var = E[(X - E[X])**2]
где X — случайная величина, связанная с этим распределением, E обозначает математическое ожидание, и Var.shape = batch_shape + event_shape.
| Аргументы | |
|---|---|
name | Python-строка, добавляемая к именам созданных этой функцией операций. |
| Возвращаемое значение | |
|---|---|
variance | Вещественное число Tensor с формой, идентичной batch_shape + event_shape, т.е. такой же формой, как у self.mean(). |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/compat/v1/distributions/Distribution