mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-05 07:04:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Small change to avoid hardcoded change in make_kn_lm.py

Browse Source 
Signed-off-by: Xinyuan Li <xli257@b17.clsp.jhu.edu>
This commit is contained in:
Xinyuan Li 2024-01-24 23:38:35 -05:00
parent eec59410f1
commit 6ba1e63c41
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
icefall/shared/make_kn_lm.py
View File

@ -169,7 +169,10 @@ class NgramCounts:
with open(filename, encoding=default_encoding) as fp: with open(filename, encoding=default_encoding) as fp:
for line in fp: for line in fp:
line = line.strip(strip_chars) line = line.strip(strip_chars)
self.add_raw_counts_from_line(line.split()[0]) if self.ngram_order == 1:
self.add_raw_counts_from_line(line.split()[0])
else:
self.add_raw_counts_from_line(line)
lines_processed += 1 lines_processed += 1
if lines_processed == 0 or args.verbose > 0: if lines_processed == 0 or args.verbose > 0:
print( print(