def gen_test():
print("### HMM Candidates Test")
hmm = HMM(1, max_edits=1, max_states=5)
hmm.train(words_ds="../data/word_freq/lotr_language_model.txt",
sentences_ds="../data/texts/lotr_clean.txt",
typo_ds="../data/typo/clean/lotr_train.csv")
pp.pprint("Typed: andd")
start = time.time()
x = hmm.candidates("andd")
end = time.time()
pred_time = end - start
pp.pprint("Time: " + str(pred_time))
pp.pprint(x)
def hmm_candidate_test():
print("### HMM Candidates Test")
hmm = HMM(1, max_edits=2, max_states=5)
hmm.train(words_ds="../data/word_freq/lotr_language_model.txt",
sentences_ds="../data/texts/lotr_clean.txt",
typo_ds="../data/typo/clean/lotr_train.csv")
pp.pprint("Typed: bools")
start = time.time()
x = hmm.candidates("bools")
end = time.time()
pred_time = end - start
pp.pprint("Time: " + str(pred_time))
pp.pprint(x)
pp.pprint("Typed: peculair")
start = time.time()
x = hmm.candidates("peculair")
end = time.time()
pred_time = end - start
pp.pprint("Time: " + str(pred_time))
pp.pprint(x)
pp.pprint("Typed: migt")
start = time.time()
x = hmm.candidates("migt")
end = time.time()
pred_time = end - start
pp.pprint("Time: " + str(pred_time))
pp.pprint(x)
pp.pprint("Typed: littele")
start = time.time()
x = hmm.candidates("littele")
end = time.time()
pred_time = end - start
pp.pprint("Time: " + str(pred_time))
pp.pprint(x)
pp.pprint("Typed: hoem")
start = time.time()
x = hmm.candidates("hoem")
end = time.time()
pred_time = end - start
pp.pprint("Time: " + str(pred_time))
pp.pprint(x)
pp.pprint("Typed: tome")
start = time.time()
x = hmm.candidates("tome")
end = time.time()
pred_time = end - start
pp.pprint("Time: " + str(pred_time))
pp.pprint(x)
pp.pprint("Typed: permforance")
start = time.time()
x = hmm.candidates("permforance")
end = time.time()
pred_time = end - start
pp.pprint("Time: " + str(pred_time))
pp.pprint(x)
pp.pprint("Typed: storry")
start = time.time()
x = hmm.candidates("storry")
end = time.time()
pred_time = end - start
pp.pprint("Time: " + str(pred_time))
pp.pprint(x)
print("\n")
