parent
6bc445f235
commit
37b236869f
@ -1,22 +0,0 @@
|
||||
"""Contains various CTC decoders in SWIG."""
|
||||
from __future__ import absolute_import
|
||||
from __future__ import division
|
||||
from __future__ import print_function
|
||||
|
||||
from swig_ctc_beam_search_decoder import ctc_beam_search_decoder as beam_search_decoder
|
||||
from swig_ctc_beam_search_decoder import ctc_best_path_decoder as best_path__decoder
|
||||
|
||||
|
||||
def ctc_best_path_decoder(probs_seq, vocabulary):
|
||||
best_path__decoder(probs_seq.to_list(), vocabulary)
|
||||
|
||||
|
||||
def ctc_beam_search_decoder(
|
||||
probs_seq,
|
||||
beam_size,
|
||||
vocabulary,
|
||||
blank_id,
|
||||
cutoff_prob=1.0,
|
||||
ext_scoring_func=None, ):
|
||||
beam_search_decoder(probs_seq.to_list(), beam_size, vocabulary, blank_id,
|
||||
cutoff_prob, ext_scoring_func)
|
Loading…
Reference in new issue