tf.random.learned_unigram_candidate_sampler

Выборка набора классов из распределения, обученного во время обучения.

Просмотр псевдонимов

Основные псевдонимы

tf.nn.learned_unigram_candidate_sampler

Псевдонимы совместимости для миграции

См. Руководство по миграции для получения более подробной информации.

tf.compat.v1.nn.learned_unigram_candidate_sampler, tf.compat.v1.random.learned_unigram_candidate_sampler

tf.random.learned_unigram_candidate_sampler(
    true_classes,
    num_true,
    num_sampled,
    unique,
    range_max,
    seed=None,
    name=None
)

Эта операция случайным образом выбирает тензор выборочных классов (sampled_candidates) из диапазона целых чисел [0, range_max).

Элементы sampled_candidates выбираются без замены (если unique=True) или с заменой (если unique=False) из базового распределения.

Базовое распределение для этой операции строится на лету во время обучения. Это распределение униграмм по целям, которые наблюдались до сих пор во время обучения. Каждое целое число в [0, range_max) начинается с веса 1 и увеличивается на 1 каждый раз, когда оно встречается в качестве целевого класса. Базовое распределение не сохраняется в контрольных точках, поэтому оно сбрасывается при повторной загрузке модели.

Кроме того, эта операция возвращает тензоры true_expected_count и sampled_expected_count, представляющие количество раз, когда каждый из целевых классов (true_classes) и выборочных классов (sampled_candidates) ожидается в среднем тензоре выборочных классов. Эти значения соответствуют Q(y|x) в документе этот документ. Если unique=True, то это вероятности после отбраковки, и мы вычисляем их приближённо.

Аргументы
`true_classes`	A `Tensor` типа `int64` и формы `[batch_size, num_true]`. Целевые классы.
`num_true`	An `int`. Количество целевых классов на пример обучения.
`num_sampled`	An `int`. Количество классов, подлежащих случайной выборке.
`unique`	A `bool`. Определяет, уникальны ли все выборочные классы в наборе.
`range_max`	An `int`. Количество возможных классов.
`seed`	An `int`. Семена, специфичные для операции. По умолчанию 0.
`name`	Имя операции (необязательно).

Возвращаемые значения
`sampled_candidates`	A tensor of type `int64` and shape `[num_sampled]`. Выборочные классы.
`true_expected_count`	A tensor of type `float`. Форма такая же, как у `true_classes`. Ожидаемые подсчёты в рамках распределения выборки для каждого из `true_classes`.
`sampled_expected_count`	A tensor of type `float`. Форма такая же, как у `sampled_candidates`. Ожидаемые подсчёты в рамках распределения выборки для каждого из `sampled_candidates`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/random/learned_unigram_candidate_sampler