bayes_test.py 文件源码

def load_prior_prob(file_name):
    f_input = open(file_name,"r")
    is_first_line = 0
    try:
        while True:
            line = f_input.readline()
            line = line.strip()
            if(len(line) == 0):
                break
            items = line.split("\t")
            if len(items) != 3:
                print "error in the prob file!",len(items),line
                sys.exit()
            if is_first_line == 0:
                sign_id_map[items[1]] = 0
                sign_id_map[items[2]] = 1
                is_first_line = 1
            else:
                word_list[items[0]] = len(prob_matrix)
                prob_matrix.append((string.atof(items[1]),string.atof(items[2]),items[0]))
            #print line
    finally:
        for i in range(len(prob_matrix)):
            if word_list[prob_matrix[i][2]] != i:
                print "load prob error !"
                sys.exit()
        f_input.close()