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))