treat unk as blank (#299)

2023-09-07 15:12:29 +08:00
parent ffeff3b8a3
commit a12ebfab22
5 changed files with 29 additions and 12 deletions
--- a/sherpa-onnx/csrc/online-transducer-modified-beam-search-decoder.h
+++ b/sherpa-onnx/csrc/online-transducer-modified-beam-search-decoder.h
@@ -21,11 +21,12 @@ class OnlineTransducerModifiedBeamSearchDecoder
  OnlineTransducerModifiedBeamSearchDecoder(OnlineTransducerModel *model,
                                            OnlineLM *lm,
                                            int32_t max_active_paths,
-                                            float lm_scale)
+                                            float lm_scale, int32_t unk_id)
      : model_(model),
        lm_(lm),
        max_active_paths_(max_active_paths),
-        lm_scale_(lm_scale) {}
+        lm_scale_(lm_scale),
+        unk_id_(unk_id) {}

  OnlineTransducerDecoderResult GetEmptyResult() const override;

@@ -45,6 +46,7 @@ class OnlineTransducerModifiedBeamSearchDecoder

  int32_t max_active_paths_;
  float lm_scale_;  // used only when lm_ is not nullptr
+  int32_t unk_id_;
 };

 }  // namespace sherpa_onnx