crf_decode: CRF decode

View source: R/text_ops.R

crf_decodeR Documentation

CRF decode

Description

Decode the highest scoring sequence of tags.

Usage

crf_decode(potentials, transition_params, sequence_length)

Arguments

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.

Value

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


henry090/tfaddons documentation built on April 7, 2022, 11:27 p.m.