Module: tfm.nlp.ops

Ops package definition.

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.