tfa.text.viterbi_decode

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

Decode the highest scoring sequence of tags outside of TensorFlow.

This should only be used at test time.

score A [seq_len, num_tags] matrix of unary potentials.
transition_params A [num_tags, num_tags] matrix of binary potentials.

viterbi A [seq_len] list of integers containing the highest scoring tag indices.
viterbi_score A float containing the score for the Viterbi sequence.