Thanks for tuning in to Google I/O. View all sessions on demandWatch on demand

Module: tfm.nlp.ops

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.