tf.raw_ops.AllCandidateSampler
Генерирует метки для выборки кандидатов с помощью обученной униграмной распределения.
tf.raw_ops.AllCandidateSampler(
true_classes, num_true, num_sampled, unique, seed=0, seed2=0, name=None
)
См. объяснения выборки кандидатов и форматов данных по адресу go/candidate-sampling.
Для каждой группы эта операция выбирает единственный набор выборочных меток кандидатов.
Преимущества выборки кандидатов на уровне группы — простота и возможность эффективного умножения плотной матрицы. Недостаток состоит в том, что выбранные кандидаты должны выбираться независимо от контекста и истинных меток.
| Аргументы | |
|---|---|
true_classes | A Tensor типа int64. Матрица размера batch_size * num_true, в которой каждая строка содержит идентификаторы num_true целевых классов в соответствующей исходной метке. |
num_true | Целое число, представляющее собой >= 1. Количество истинных меток на контекст. |
num_sampled | Целое число, представляющее собой >= 1. Количество кандидатов для получения. |
unique | Булево значение. Если unique истинно, мы осуществляем выборку с отвержением, чтобы все выбранные кандидаты в группе были уникальны. Это требует некоторого приближения для оценки вероятностей выборки после отвержения. |
seed | Необязательное целое число. По умолчанию 0. Если seed или seed2 установлены не равными нулю, генератор случайных чисел инициализируется заданным значением seed. В противном случае он инициализируется случайным значением seed. |
seed2 | Необязательное целое число. По умолчанию 0. Второй seed для предотвращения конфликтов seed. |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
Кортеж из Tensor объектов (sampled_candidates, true_expected_count, sampled_expected_count). | |
sampled_candidates | A Tensor типа int64. |
true_expected_count | A Tensor типа float32. |
sampled_expected_count | A Tensor типа float32. |
© 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/raw_ops/AllCandidateSampler
