tfa.text.viterbi_decode

Decode the highest scoring sequence of tags outside of TensorFlow.

View aliases

Main aliases

tfa.text.crf.viterbi_decode

tfa.text.viterbi_decode(
    score: tfa.types.TensorLike,
    transition_params: tfa.types.TensorLike
) -> tf.Tensor

This should only be used at test time.

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

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