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.crf_decode

View source on GitHub

Decode the highest scoring sequence of tags in TensorFlow.

Aliases:

tfa.text.crf_decode(
    potentials,
    transition_params,
    sequence_length
)

This is a function for tensor.

Args:

  • potentials: A [batch_size, max_seq_len, num_tags] tensor of unary potentials.
  • transition_params: A [num_tags, num_tags] matrix of binary potentials.
  • sequence_length: A [batch_size] vector of true sequence lengths.

Returns:

  • decode_tags: A [batch_size, max_seq_len] matrix, with dtype tf.int32. Contains the highest scoring tag indices.
  • best_score: A [batch_size] vector, containing the score of decode_tags.