ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Candidate Sampling Ops

Summary

Classes

tensorflow:: ops:: AllCandidateSampler

Generates labels for candidate sampling with a learned unigram distribution.

tensorflow:: ops:: ComputeAccidentalHits

Computes the ids of the positions in sampled_candidates that match true_labels.

tensorflow:: ops:: FixedUnigramCandidateSampler

Generates labels for candidate sampling with a learned unigram distribution.

tensorflow:: ops:: LearnedUnigramCandidateSampler

Generates labels for candidate sampling with a learned unigram distribution.

tensorflow:: ops:: LogUniformCandidateSampler

Generates labels for candidate sampling with a log-uniform distribution.

tensorflow:: ops:: UniformCandidateSampler

Generates labels for candidate sampling with a uniform distribution.