mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-10 01:24:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

enable sclite_mode for swbd scoring

Browse Source
This commit is contained in:
zr_jin 2023-09-04 10:56:22 +08:00
parent fc2df07841
commit ecbc026061
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
icefall/utils.py
View File

@ -493,6 +493,7 @@ def write_error_stats(
test_set_name: str, test_set_name: str,
results: List[Tuple[str, str]], results: List[Tuple[str, str]],
enable_log: bool = True, enable_log: bool = True,
sclite_mode: bool = False,
) -> float: ) -> float:
"""Write statistics based on predicted results and reference transcripts. """Write statistics based on predicted results and reference transcripts.
@ -538,7 +539,7 @@ def write_error_stats(
num_corr = 0 num_corr = 0
ERR = "*" ERR = "*"
for cut_id, ref, hyp in results: for cut_id, ref, hyp in results:
ali = kaldialign.align(ref, hyp, ERR) ali = kaldialign.align(ref, hyp, ERR, sclite_mode=sclite_mode)
for ref_word, hyp_word in ali: for ref_word, hyp_word in ali:
if ref_word == ERR: if ref_word == ERR:
ins[hyp_word] += 1 ins[hyp_word] += 1