def test_calc_vector_MLE(): """Tests the Trainer.calc_vector_MLE function.""" print("\tTesting Trainer.calc_vector_MLE()...") t = Trainer() MLE_vec = t.calc_vector_MLE() fname = DEFAULT_TEST_PATH + "test_calc_vector_MLE.txt" test_file = open(fname, "w") print(MLE_vec, file=test_file) test_file.close()
def test_calc_matrix_MAP(): """Tests the Trainer.calc_matrix_MAP function.""" print("\tTesting Trainer.calc_matrix_MAP()...") t = Trainer() MLE_vec = t.calc_vector_MLE() #required because it sets labeldict. MAP_matrix = t.calc_matrix_MAP() fname = DEFAULT_TEST_PATH + "test_calc_matrix_MAP.txt" test_f = open(fname, "w") print(MAP_matrix, file=test_f) print("shape:{}".format(MAP_matrix.shape), file=test_f) print("type:{}".format(MAP_matrix.dtype), file=test_f) print("min:{}".format(MAP_matrix.min()), file=test_f) print("max:{}".format(MAP_matrix.max()), file=test_f) test_f.close()
def test_word_ranking(): t = Trainer() MLE_vec = t.calc_vector_MLE() MAP_matrix = t.calc_matrix_MAP() t.get_word_ranking(MAP_matrix, MLE_vec)