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.