update decoder py3

pull/519/head
Hui Zhang 5 years ago
parent 00e109752a
commit 862f9dec34

@ -18,10 +18,10 @@ addons:
- python3-dev - python3-dev
before_install: before_install:
- python --version - sudo pip3 install -U virtualenv pre-commit pip
- pip --version
- sudo pip install -U virtualenv pre-commit pip
- docker pull paddlepaddle/paddle:latest - docker pull paddlepaddle/paddle:latest
- python3 --version
- pip3 --version
script: script:
- exit_code=0 - exit_code=0

@ -120,7 +120,7 @@ def ctc_beam_search_decoder(probs_seq,
prob_idx = prob_idx[0:cutoff_len] prob_idx = prob_idx[0:cutoff_len]
for l in prefix_set_prev: for l in prefix_set_prev:
if not prefix_set_next.has_key(l): if l not in prefix_set_next:
probs_b_cur[l], probs_nb_cur[l] = 0.0, 0.0 probs_b_cur[l], probs_nb_cur[l] = 0.0, 0.0
# extend prefix by travering prob_idx # extend prefix by travering prob_idx
@ -134,7 +134,7 @@ def ctc_beam_search_decoder(probs_seq,
last_char = l[-1] last_char = l[-1]
new_char = vocabulary[c] new_char = vocabulary[c]
l_plus = l + new_char l_plus = l + new_char
if not prefix_set_next.has_key(l_plus): if l_plus not in prefix_set_next:
probs_b_cur[l_plus], probs_nb_cur[l_plus] = 0.0, 0.0 probs_b_cur[l_plus], probs_nb_cur[l_plus] = 0.0, 0.0
if new_char == last_char: if new_char == last_char:
@ -161,7 +161,7 @@ def ctc_beam_search_decoder(probs_seq,
## store top beam_size prefixes ## store top beam_size prefixes
prefix_set_prev = sorted( prefix_set_prev = sorted(
prefix_set_next.iteritems(), key=lambda asd: asd[1], reverse=True) prefix_set_next.items(), key=lambda asd: asd[1], reverse=True)
if beam_size < len(prefix_set_prev): if beam_size < len(prefix_set_prev):
prefix_set_prev = prefix_set_prev[:beam_size] prefix_set_prev = prefix_set_prev[:beam_size]
prefix_set_prev = dict(prefix_set_prev) prefix_set_prev = dict(prefix_set_prev)

Loading…
Cancel
Save