@@ -329,11 +329,10 @@ <h1>Source code for decoders.fc_decoders</h1><div class="highlight"><pre>
329329 < span class ="k "> return</ span > < span class ="nb "> dict</ span > < span class ="p "> (</ span > < span class ="n "> FullyConnectedTimeDecoder</ span > < span class ="o "> .</ span > < span class ="n "> get_optional_params</ span > < span class ="p "> (),</ span > < span class ="o "> **</ span > < span class ="p "> {</ span >
330330 < span class ="s1 "> 'decoder_library_path'</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="p "> ,</ span >
331331 < span class ="s1 "> 'beam_width'</ span > < span class ="p "> :</ span > < span class ="nb "> int</ span > < span class ="p "> ,</ span >
332- < span class ="s1 "> 'lm_weight'</ span > < span class ="p "> :</ span > < span class ="nb "> float</ span > < span class ="p "> ,</ span >
333- < span class ="s1 "> 'word_count_weight'</ span > < span class ="p "> :</ span > < span class ="nb "> float</ span > < span class ="p "> ,</ span >
334- < span class ="s1 "> 'valid_word_count_weight'</ span > < span class ="p "> :</ span > < span class ="nb "> float</ span > < span class ="p "> ,</ span >
335- < span class ="s1 "> 'lm_binary_path'</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="p "> ,</ span >
336- < span class ="s1 "> 'lm_trie_path'</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="p "> ,</ span >
332+ < span class ="s1 "> 'alpha'</ span > < span class ="p "> :</ span > < span class ="nb "> float</ span > < span class ="p "> ,</ span >
333+ < span class ="s1 "> 'beta'</ span > < span class ="p "> :</ span > < span class ="nb "> float</ span > < span class ="p "> ,</ span >
334+ < span class ="s1 "> 'lm_path'</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="p "> ,</ span >
335+ < span class ="s1 "> 'trie_path'</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="p "> ,</ span >
337336 < span class ="s1 "> 'alphabet_config_path'</ span > < span class ="p "> :</ span > < span class ="nb "> str</ span > < span class ="p "> ,</ span >
338337 < span class ="p "> })</ span > </ div >
339338
@@ -349,16 +348,14 @@ <h1>Source code for decoders.fc_decoders</h1><div class="highlight"><pre>
349348< span class ="sd "> output text generation. If False, other config parameters are not used.</ span >
350349< span class ="sd "> * **decoder_library_path** (string) --- path to the ctc decoder with</ span >
351350< span class ="sd "> language model library.</ span >
352- < span class ="sd "> * **lm_binary_path ** (string) --- path to the language model file.</ span >
353- < span class ="sd "> * **lm_trie_path ** (string) --- path to the language model trie file.</ span >
351+ < span class ="sd "> * **lm_path ** (string) --- path to the language model file.</ span >
352+ < span class ="sd "> * **trie_path ** (string) --- path to the prefix trie file.</ span >
354353< span class ="sd "> * **alphabet_config_path** (string) --- path to the alphabet file.</ span >
355354< span class ="sd "> * **beam_width** (int) --- beam width for beam search.</ span >
356- < span class ="sd "> * **lm_weight ** (float) --- weight that is assigned to language model</ span >
355+ < span class ="sd "> * **alpha ** (float) --- weight that is assigned to language model</ span >
357356< span class ="sd "> probabilities.</ span >
358- < span class ="sd "> * **word_count_weight ** (float) --- weight that is assigned to the</ span >
357+ < span class ="sd "> * **beta ** (float) --- weight that is assigned to the</ span >
359358< span class ="sd "> word count.</ span >
360- < span class ="sd "> * **valid_word_count_weight** (float) --- weight that is assigned to the</ span >
361- < span class ="sd "> valid word count, i.e. words that exist in language model dictionary.</ span >
362359< span class ="sd "> """</ span >
363360 < span class ="nb "> super</ span > < span class ="p "> (</ span > < span class ="n "> FullyConnectedCTCDecoder</ span > < span class ="p "> ,</ span > < span class ="bp "> self</ span > < span class ="p "> )</ span > < span class ="o "> .</ span > < span class ="fm "> __init__</ span > < span class ="p "> (</ span > < span class ="n "> params</ span > < span class ="p "> ,</ span > < span class ="n "> model</ span > < span class ="p "> ,</ span > < span class ="n "> name</ span > < span class ="p "> ,</ span > < span class ="n "> mode</ span > < span class ="p "> )</ span >
364361
@@ -382,12 +379,10 @@ <h1>Source code for decoders.fc_decoders</h1><div class="highlight"><pre>
382379 < span class ="n "> decoded_ixs</ span > < span class ="p "> ,</ span > < span class ="n "> decoded_vals</ span > < span class ="p "> ,</ span > < span class ="n "> decoded_shapes</ span > < span class ="p "> ,</ span > < span class ="n "> log_probabilities</ span > < span class ="o "> =</ span > < span class ="p "> (</ span >
383380 < span class ="n "> custom_op_module</ span > < span class ="o "> .</ span > < span class ="n "> ctc_beam_search_decoder_with_lm</ span > < span class ="p "> (</ span >
384381 < span class ="n "> logits</ span > < span class ="p "> ,</ span > < span class ="n "> sequence_length</ span > < span class ="p "> ,</ span > < span class ="n "> beam_width</ span > < span class ="o "> =</ span > < span class ="n "> beam_width</ span > < span class ="p "> ,</ span >
385- < span class ="n "> model_path</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> params</ span > < span class ="p "> [</ span > < span class ="s1 "> 'lm_binary_path'</ span > < span class ="p "> ],</ span >
386- < span class ="n "> trie_path</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> params</ span > < span class ="p "> [</ span > < span class ="s1 "> 'lm_trie_path'</ span > < span class ="p "> ],</ span >
382+ < span class ="n "> model_path</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> params</ span > < span class ="p "> [</ span > < span class ="s1 "> 'lm_path'</ span > < span class ="p "> ],</ span > < span class ="n "> trie_path</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> params</ span > < span class ="p "> [</ span > < span class ="s1 "> 'trie_path'</ span > < span class ="p "> ],</ span >
387383 < span class ="n "> alphabet_path</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> params</ span > < span class ="p "> [</ span > < span class ="s1 "> 'alphabet_config_path'</ span > < span class ="p "> ],</ span >
388- < span class ="n "> lm_weight</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> params</ span > < span class ="p "> [</ span > < span class ="s1 "> 'lm_weight'</ span > < span class ="p "> ],</ span >
389- < span class ="n "> word_count_weight</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> params</ span > < span class ="p "> [</ span > < span class ="s1 "> 'word_count_weight'</ span > < span class ="p "> ],</ span >
390- < span class ="n "> valid_word_count_weight</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> params</ span > < span class ="p "> [</ span > < span class ="s1 "> 'valid_word_count_weight'</ span > < span class ="p "> ],</ span >
384+ < span class ="n "> alpha</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> params</ span > < span class ="p "> [</ span > < span class ="s1 "> 'alpha'</ span > < span class ="p "> ],</ span >
385+ < span class ="n "> beta</ span > < span class ="o "> =</ span > < span class ="bp "> self</ span > < span class ="o "> .</ span > < span class ="n "> params</ span > < span class ="p "> [</ span > < span class ="s1 "> 'beta'</ span > < span class ="p "> ],</ span >
391386 < span class ="n "> top_paths</ span > < span class ="o "> =</ span > < span class ="n "> top_paths</ span > < span class ="p "> ,</ span > < span class ="n "> merge_repeated</ span > < span class ="o "> =</ span > < span class ="n "> merge_repeated</ span > < span class ="p "> ,</ span >
392387 < span class ="p "> )</ span >
393388 < span class ="p "> )</ span >
0 commit comments