mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-08 08:34: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,
results: List[Tuple[str, str]],
enable_log: bool = True,
sclite_mode: bool = False,
) -> float:
"""Write statistics based on predicted results and reference transcripts.
@ -538,7 +539,7 @@ def write_error_stats(
num_corr = 0
ERR = "*"
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:
if ref_word == ERR:
ins[hyp_word] += 1