예제 #1
0
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)
예제 #2
0
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")