From 904a3cc8a9fb396be39a456392b79ebef2af50de Mon Sep 17 00:00:00 2001
From: AHN Sung Hwan <aask1357@naver.com>
Date: Thu, 11 Apr 2024 11:34:44 +0900
Subject: [PATCH] Fix a bug in mean calculation of 'ys_probs' (#748)

---
 sherpa-onnx/csrc/transducer-keyword-decoder.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sherpa-onnx/csrc/transducer-keyword-decoder.cc b/sherpa-onnx/csrc/transducer-keyword-decoder.cc
index f31348ea..af78cb9c 100644
--- a/sherpa-onnx/csrc/transducer-keyword-decoder.cc
+++ b/sherpa-onnx/csrc/transducer-keyword-decoder.cc
@@ -152,7 +152,7 @@ void TransducerKeywordDecoder::Decode(
       if (matched) {
         float ys_prob = 0.0;
         int32_t length = best_hyp.ys_probs.size();
-        for (int32_t i = 1; i <= matched_state->level; ++i) {
+        for (int32_t i = 0; i < matched_state->level; ++i) {
           ys_prob += best_hyp.ys_probs[i];
         }
         ys_prob /= matched_state->level;