def decode_bigram(pinyins):
"""
Use Viterbi to decode the pinyin sequence.
"""
lm = LanguageModel.load_from_trained()
for pinyin in pinyins:
print(pinyin)
print(viterbi_bigram(lm, pinyin))
def decode_trigram(pinyins):
"""
Use Viterbi to decode the pinyin sequence.
"""
lm = LanguageModel.load_from_trained(model_path="models/3-lm.pkl")
print("load {}-gram model successfully.".format(lm.ngram))
for pinyin in pinyins:
print(pinyin)
print(viterbi_trigram(lm, pinyin))
