Module: tfm.nlp.ops

Stay organized with collections Save and categorize content based on your preferences.

Ops package definition.


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

class SequenceBeamSearch: Implementation of beam search loop.


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.