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

Merge pull request #3021 from mozilla/confidence-raw-scores

Browse Source 
Return raw scores in confidence value (Fixes #3004)
This commit is contained in:
Reuben Morais 2020-05-27 18:40:02 +02:00 committed by GitHub
commit af4bc31c27
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
native_client/ctcdecode/ctc_beam_search_decoder.cpp
View File

@ -204,20 +204,10 @@ DecoderState::decode(size_t num_results) const
std::vector<Output> outputs;
outputs.reserve(num_returned);
// compute aproximate ctc score as the return score, without affecting the
// return order of decoding result. To delete when decoder gets stable.
for (size_t i = 0; i < num_returned; ++i) {
Output output;
prefixes_copy[i]->get_path_vec(output.tokens, output.timesteps);
double approx_ctc = scores[prefixes_copy[i]];
if (ext_scorer_) {
auto words = ext_scorer_->split_labels_into_scored_units(output.tokens);
// remove term insertion weight
approx_ctc -= words.size() * ext_scorer_->beta;
// remove language model weight
approx_ctc -= (ext_scorer_->get_sent_log_prob(words)) * ext_scorer_->alpha;
}
output.confidence = -approx_ctc;
output.confidence = scores[prefixes_copy[i]];
outputs.push_back(output);
}