Module: tfm.nlp.ops

Ops package definition.

Classes

class SamplingModule: Implementation for sampling strategies (go/decoding-tf-nlp).

class SequenceBeamSearch: Implementation of beam search loop.

Functions

get_next_sentence_labels(...): Extracts the next sentence label from sentences.

get_sentence_order_labels(...): Extract segments and labels for sentence order prediction (SOP) task.

sequence_beam_search(...): Search for sequence of subtoken ids with the largest probability.