TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tfa.text.viterbi_decode

View source on GitHub

Decode the highest scoring sequence of tags outside of TensorFlow.

Aliases:

tfa.text.viterbi_decode(
    score,
    transition_params
)

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.