TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

Module: tfa.text.skip_gram_ops

View source on GitHub

Skip-gram sampling ops from https://arxiv.org/abs/1301.3781.

Functions

skip_gram_sample(...): Generates skip-gram token and label paired Tensors from the input

skip_gram_sample_with_text_vocab(...): Skip-gram sampling with a text vocabulary file.