tf.random.learned_unigram_candidate_sampler
| Просмотр исходного кода на GitHub |
Выборка набора классов из распределения, обученного во время обучения.
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 типа int64 и формы [num_sampled]. Выбранные классы. |
true_expected_count | A tensor типа float. Такая же форма, как у true_classes. Ожидаемое количество по распределению выборки каждого из true_classes. |
sampled_expected_count | A tensor типа float. Такая же форма, как у sampled_candidates. Ожидаемое количество по распределению выборки каждого из sampled_candidates. |
© 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/r1.15/api_docs/python/tf/random/learned_unigram_candidate_sampler
