Fix in the confidence ensemble test (#7682)

* Fix in the confidence ensemble test Signed-off-by: Igor Gitman <[email protected]> * Correct parameter names Signed-off-by: Igor Gitman <[email protected]> --------- Signed-off-by: Igor Gitman <[email protected]>
NVIDIA · Oct 13, 2023 · 33d04b2 · 33d04b2
1 parent 188f0a1
commit 33d04b2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/scripts/confidence_ensembles/build_ensemble.py b/scripts/confidence_ensembles/build_ensemble.py
@@ -215,7 +215,7 @@ class BuildEnsembleConfig:
             preserve_frame_confidence=True,
             exclude_blank=True,
             aggregation="mean",
-            measure_cfg=ConfidenceMethodConfig(name="entropy", entropy_type="renyi", alpha=0.25, entropy_norm="lin",),
+            method_cfg=ConfidenceMethodConfig(name="entropy", entropy_type="renyi", alpha=0.25, entropy_norm="lin",),
         )
     )
     temperature: float = 1.0

diff --git a/scripts/confidence_ensembles/test_confidence_ensembles.py b/scripts/confidence_ensembles/test_confidence_ensembles.py
@@ -33,7 +33,10 @@
     [
         "ensemble.0.model=stt_es_conformer_ctc_large ensemble.1.model=stt_it_conformer_ctc_large",
         "ensemble.0.model=stt_es_conformer_transducer_large ensemble.1.model=stt_it_conformer_transducer_large",
-        "ensemble.0.model=stt_es_fastconformer_hybrid_large_pc ensemble.1.model=stt_it_fastconformer_hybrid_large_pc",
+        (
+            "ensemble.0.model=stt_es_fastconformer_hybrid_large_pc ensemble.1.model=stt_it_fastconformer_hybrid_large_pc "
+            "confidence.method_cfg.alpha=0.33 confidence.method_cfg.entropy_norm=exp "
+        ),
         (
             "ensemble.0.model=stt_es_fastconformer_hybrid_large_pc "
             "ensemble.1.model=stt_it_fastconformer_hybrid_large_pc "