Skip to content

Commit 835fa69

Browse files
committed
rename get_language_model -> make_bigram_language_model
1 parent 71be2db commit 835fa69

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

gentle/language_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def get_node_id(word):
4949

5050
return output
5151

52-
def get_language_model(kaldi_seq, proto_langdir='PROTO_LANGDIR'):
52+
def make_bigram_language_model(kaldi_seq, proto_langdir='PROTO_LANGDIR'):
5353
"""Generates a language model to fit the text.
5454
5555
Returns the filename of the generated language model FST.
@@ -83,4 +83,4 @@ def get_language_model(kaldi_seq, proto_langdir='PROTO_LANGDIR'):
8383

8484
if __name__=='__main__':
8585
import sys
86-
get_language_model(open(sys.argv[1]).read())
86+
make_bigram_language_model(open(sys.argv[1]).read())

gentle/language_model_transcribe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ def lm_transcribe(audio_f, transcript, proto_langdir, nnet_dir,
2727

2828
ks = ms.get_kaldi_sequence()
2929

30-
gen_hclg_filename = language_model.get_language_model(ks, proto_langdir)
30+
gen_hclg_filename = language_model.make_bigram_language_model(ks, proto_langdir)
3131
try:
3232
k = standard_kaldi.Kaldi(nnet_dir, gen_hclg_filename, proto_langdir)
3333

0 commit comments

Comments
 (0)