experiments/STT
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Don't OOV_SCORE on empty prefix

Browse Source 
Fixes #2579
This commit is contained in:
Alexandre Lissy 2019-12-17 09:39:14 +01:00
parent 6fa2babdfd
commit f44d2ddeb9
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
native_client/ctcdecode/ctc_beam_search_decoder.cpp
View File

@ -169,9 +169,7 @@ DecoderState::decode() const
if (ext_scorer_ != nullptr) { if (ext_scorer_ != nullptr) {
for (size_t i = 0; i < beam_size_ && i < prefixes_copy.size(); ++i) { for (size_t i = 0; i < beam_size_ && i < prefixes_copy.size(); ++i) {
auto prefix = prefixes_copy[i]; auto prefix = prefixes_copy[i];
if (prefix->is_empty()) { if (!ext_scorer_->is_scoring_boundary(prefix->parent, prefix->character)) {
scores[prefix] = OOV_SCORE;
} else if (!ext_scorer_->is_scoring_boundary(prefix->parent, prefix->character)) {
float score = 0.0; float score = 0.0;
std::vector<std::string> ngram = ext_scorer_->make_ngram(prefix); std::vector<std::string> ngram = ext_scorer_->make_ngram(prefix);
bool bos = ngram.size() < ext_scorer_->get_max_order(); bool bos = ngram.size() < ext_scorer_->get_max_order();