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()

Методы

batch_shape_tensor

Просмотреть исходный код

batch_shape_tensor(
    name='batch_shape_tensor'
)

Форма одной выборки из одного индекса события в виде одномерного Tensor.

Измерения пакета — индексы в независимых, не идентичных параметризациях этого распределения.

cdf

Просмотреть исходный код

cdf(
    value, name='cdf'
)

Функция кумулятивного распределения.

Для случайной величины X, функция кумулятивного распределения cdf равна:

cdf(x) := P[X <= x]

copy

Просмотреть исходный код

copy(
    **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'.

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.

entropy

Просмотреть исходный код

entropy(
    name='entropy'
)

Энтропия Шеннона в натах.

event_shape_tensor

Просмотреть исходный код

event_shape_tensor(
    name='event_shape_tensor'
)

Форма отдельной выборки из отдельной партии как 1-мерный целочисленный Tensor.

is_scalar_batch

Просмотреть исходный код

is_scalar_batch(
    name='is_scalar_batch'
)

Указывает, что batch_shape == [].

is_scalar_event

Просмотреть исходный код

is_scalar_event(
    name='is_scalar_event'
)

Указывает, что event_shape == [].

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[.] обозначает (Шаннон) энтропию.

log_cdf

Просмотреть исходный код

log_cdf(
    value, name='log_cdf'
)

Логарифм функции кумулятивного распределения.

Для случайной величины X, функция кумулятивного распределения cdf определяется как:

log_cdf(x) := Log[ P[X <= x] ]

Часто для log_cdf(x) используется численное приближение, которое даёт более точный результат, чем просто взятие логарифма cdf в случае x << -1.

log_prob

Просмотреть исходный код

log_prob(
    value, name='log_prob'
)

Логарифм плотности/массы вероятности.

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.

Среднее значение

Просмотреть исходный код

mean(
    name='mean'
)

Среднее значение.

Мода

Просмотреть исходный код

mode(
    name='mode'
)

Мода.

Формы параметров

Просмотреть исходный код

@classmethod
param_shapes(
    sample_shape, name='DistributionParamShapes'
)

Формы параметров, учитывая желаемую форму вызова sample().

Это метод класса, который описывает, какие пары ключ/значение требуются для создания данного Distribution, чтобы получить определённую форму результата вызова sample() данным экземпляром.

Подклассы должны переопределить метод класса _param_shapes.

Статические формы параметров

Просмотреть исходный код

@classmethod
param_static_shapes(
    sample_shape
)

param_shapes со статическими (т.е. TensorShape) формами.

Это метод класса, который описывает, какие пары ключ/значение требуются для создания данного Distribution, чтобы получить определённую форму результата вызова sample() данным экземпляром. Предполагается, что форма выборки известна статически.

Подклассы должны переопределить метод класса _param_shapes, чтобы возвращать тензоры с постоянными значениями при подаче постоянных значений.

Вероятность

Просмотреть исходный код

prob(
    value, name='prob'
)

Функция плотности/массы вероятности.

Квантиль

Просмотреть исходный код

quantile(
    value, name='quantile'
)

Функция квантиля. Также известна как обратная функция распределения или функция процентилей.

Для случайной переменной X и p in [0, 1], quantile равно:

quantile(p) := x such that P[X <= x] == p

Выборка

Просмотреть исходный код

sample(
    sample_shape=(), seed=None, name='sample'
)

Генерирует выборки заданной формы.

Обратите внимание, что вызов sample() без аргументов сгенерирует одну выборку.

Среднеквадратическое отклонение

Просмотреть исходный код

stddev(
    name='stddev'
)

Среднеквадратическое отклонение.

Среднеквадратическое отклонение определяется как,

stddev = E[(X - E[X])**2]**0.5

где X - случайная величина, связанная с этим распределением, E обозначает математическое ожидание, и stddev.shape = batch_shape + event_shape.

Функция выживаемости

Просмотреть исходный код

survival_function(
    value, name='survival_function'
)

Функция выживаемости.

Для случайной величины X, функция выживаемости определяется следующим образом:

survival_function(x) = P[X > x]
                     = 1 - P[X <= x]
                     = 1 - cdf(x).

Дисперсия

Просмотреть исходный код

variance(
    name='variance'
)

Дисперсия.

Дисперсия определяется как,

Var = E[(X - E[X])**2]

где X - случайная величина, связанная с этим распределением, E обозначает математическое ожидание, и Var.shape = batch_shape + event_shape.