diff options
Diffstat (limited to 'tensorflow/python/ops/candidate_sampling_ops.py')
| -rw-r--r-- | tensorflow/python/ops/candidate_sampling_ops.py | 17 |
1 files changed, 13 insertions, 4 deletions
diff --git a/tensorflow/python/ops/candidate_sampling_ops.py b/tensorflow/python/ops/candidate_sampling_ops.py index 0af257ef08..245d46b266 100644 --- a/tensorflow/python/ops/candidate_sampling_ops.py +++ b/tensorflow/python/ops/candidate_sampling_ops.py @@ -192,10 +192,19 @@ def learned_unigram_candidate_sampler(true_classes, num_true, num_sampled, seed2=seed2, name=name) -def fixed_unigram_candidate_sampler(true_classes, num_true, num_sampled, unique, - range_max, vocab_file='', distortion=1.0, - num_reserved_ids=0, num_shards=1, shard=0, - unigrams=[], seed=None, name=None): +def fixed_unigram_candidate_sampler(true_classes, + num_true, + num_sampled, + unique, + range_max, + vocab_file='', + distortion=1.0, + num_reserved_ids=0, + num_shards=1, + shard=0, + unigrams=(), + seed=None, + name=None): """Samples a set of classes using the provided (fixed) base distribution. This operation randomly samples a tensor of sampled classes |