tfa.text.crf_binary_score

Computes the binary scores of tag sequences.

View aliases

Main aliases

tfa.text.crf.crf_binary_score

tfa.text.crf_binary_score(
    tag_indices: tfa.types.TensorLike,
    sequence_lengths: tfa.types.TensorLike,
    transition_params: tfa.types.TensorLike
) -> tf.Tensor

Args
tag_indices	A [batch_size, max_seq_len] matrix of tag indices.
sequence_lengths	A [batch_size] vector of true sequence lengths.
transition_params	A [num_tags, num_tags] matrix of binary potentials.

Returns
binary_scores	A [batch_size] vector of binary scores.