get_near_word.py 文件源码

def calu_cosin_num(word_q,word_a,model):
    #print 'word_q : ' + str(word_q) + '\n'
    #print 'word_a : ' + str(word_a) + '\n'
    try:
        q_vector = model[word_q.decode('utf-8')]
        a_vector = model[word_a.decode('utf-8')]
    except KeyError:
        return 0
    cosine_similarity_num = cosine_similarity(np.array(q_vector).reshape(1,-1),np.array(a_vector).reshape(1,-1))
    return float(cosine_similarity_num)