Module: tfa.text.crf

Classes

class CrfDecodeForwardRnnCell: Computes the forward decoding in a linear-chain CRF.

Functions

crf_binary_score(...): Computes the binary scores of tag sequences.

crf_constrained_decode(...): Decode the highest scoring sequence of tags under constraints.

crf_decode(...): Decode the highest scoring sequence of tags.

crf_decode_backward(...): Computes backward decoding in a linear-chain CRF.

crf_decode_forward(...): Computes forward decoding in a linear-chain CRF.

crf_filtered_inputs(...): Constrains the inputs to filter out certain tags at each time step.

crf_forward(...): Computes the alpha values in a linear-chain CRF.

crf_log_likelihood(...): Computes the log-likelihood of tag sequences in a CRF.

crf_log_norm(...): Computes the normalization for a CRF.

crf_multitag_sequence_score(...): Computes the unnormalized score of all tag sequences matching tag_bitmap.

crf_sequence_score(...): Computes the unnormalized score for a tag sequence.

crf_unary_score(...): Computes the unary scores of tag sequences.

viterbi_decode(...): Decode the highest scoring sequence of tags outside of TensorFlow.

Type Aliases

TensorLike