Update name to ConfidenceMethodConfig

Signed-off-by: smajumdar <[email protected]>
NVIDIA · Oct 3, 2023 · 764652a · 764652a
1 parent 6af30c9
commit 764652a
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/nemo/collections/asr/parts/submodules/ctc_greedy_decoding.py b/nemo/collections/asr/parts/submodules/ctc_greedy_decoding.py
@@ -253,8 +253,8 @@ class GreedyCTCInferConfig:
     preserve_alignments: bool = False
     compute_timestamps: bool = False
     preserve_frame_confidence: bool = False
-    confidence_measure_cfg: Optional[ConfidenceMeasureConfig] = field(
-        default_factory=lambda: ConfidenceMeasureConfig()
+    confidence_measure_cfg: Optional[ConfidenceMethodConfig] = field(
+        default_factory=lambda: ConfidenceMethodConfig()
     )
     confidence_method_cfg: str = "DEPRECATED"
 

diff --git a/nemo/collections/asr/parts/submodules/rnnt_greedy_decoding.py b/nemo/collections/asr/parts/submodules/rnnt_greedy_decoding.py
@@ -2185,8 +2185,8 @@ class GreedyRNNTInferConfig:
     max_symbols_per_step: Optional[int] = 10
     preserve_alignments: bool = False
     preserve_frame_confidence: bool = False
-    confidence_measure_cfg: Optional[ConfidenceMeasureConfig] = field(
-        default_factory=lambda: ConfidenceMeasureConfig()
+    confidence_measure_cfg: Optional[ConfidenceMethodConfig] = field(
+        default_factory=lambda: ConfidenceMethodConfig()
     )
     confidence_method_cfg: str = "DEPRECATED"
 
@@ -2204,8 +2204,8 @@ class GreedyBatchedRNNTInferConfig:
     max_symbols_per_step: Optional[int] = 10
     preserve_alignments: bool = False
     preserve_frame_confidence: bool = False
-    confidence_measure_cfg: Optional[ConfidenceMeasureConfig] = field(
-        default_factory=lambda: ConfidenceMeasureConfig()
+    confidence_measure_cfg: Optional[ConfidenceMethodConfig] = field(
+        default_factory=lambda: ConfidenceMethodConfig()
     )
     confidence_method_cfg: str = "DEPRECATED"
 

diff --git a/nemo/collections/asr/parts/utils/asr_confidence_utils.py b/nemo/collections/asr/parts/utils/asr_confidence_utils.py
@@ -175,7 +175,7 @@ class ConfidenceConfig:
     preserve_word_confidence: bool = False
     exclude_blank: bool = True
     aggregation: str = "min"
-    measure_cfg: ConfidenceMeasureConfig = field(default_factory=lambda: ConfidenceMeasureConfig())
+    measure_cfg: ConfidenceMethodConfig = field(default_factory=lambda: ConfidenceMethodConfig())
     method_cfg: str = "DEPRECATED"
 
     def __post_init__(self):

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=ConfidenceMeasureConfig(name="entropy", entropy_type="renyi", alpha=0.25, entropy_norm="lin",),
+            measure_cfg=ConfidenceMethodConfig(name="entropy", entropy_type="renyi", alpha=0.25, entropy_norm="lin",),
         )
     )
     temperature: float = 1.0