tf.contrib.crf.crf_log_likelihood
Computes the log-likelihood of tag sequences in a CRF.
tf.contrib.crf.crf_log_likelihood( inputs, tag_indices, sequence_lengths, transition_params=None )
Args | |
---|---|
inputs | A [batch_size, max_seq_len, num_tags] tensor of unary potentials to use as input to the CRF layer. |
tag_indices | A [batch_size, max_seq_len] matrix of tag indices for which we compute the log-likelihood. |
sequence_lengths | A [batch_size] vector of true sequence lengths. |
transition_params | A [num_tags, num_tags] transition matrix, if available. |
Returns | |
---|---|
log_likelihood | A [batch_size] Tensor containing the log-likelihood of each example, given the sequence of tag indices. |
transition_params | A [num_tags, num_tags] transition matrix. This is either provided by the caller or created in this function. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/crf/crf_log_likelihood