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

Specify BOS for final scoring at decode if applicable

Browse Source
This commit is contained in:
Reuben Morais 2019-10-14 16:23:06 +02:00
parent c8802a38e7
commit abc687b3b9
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
native_client/ctcdecode/ctc_beam_search_decoder.cpp
View File

@ -170,7 +170,8 @@ DecoderState::decode() const
if (!prefix->is_empty() && prefix->character != space_id_) { if (!prefix->is_empty() && prefix->character != space_id_) {
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);
score = ext_scorer_->get_log_cond_prob(ngram) * ext_scorer_->alpha; bool bos = ngram.size() < ext_scorer_->get_max_order();
score = ext_scorer_->get_log_cond_prob(ngram, bos) * ext_scorer_->alpha;
score += ext_scorer_->beta; score += ext_scorer_->beta;
scores[prefix] += score; scores[prefix] += score;
} }