from?gensim.models?import?KeyedVectors

class?Word2Vec（）:
??def?__init__（self?modelPath?kind=‘bin‘）:
????“““
????創建Word2Vec對象
????
????modelPath:?模型路徑
????kind:?模型類型
??????bin:?二進制文件
??????txt:?文本文件
????return:?無
????“““
????
????if?kind?!=?‘bin‘:
??????kind?=?False
????else:
??????kind?=?True
????print（‘loading?word2vector?model...‘）
????self.model?=?KeyedVectors.load_word2vec_format（modelPath?binary=kind?unicode_errors=‘ignore‘）
??
??def?get_word_vector（self?word）:
????“““
????獲得詞向量
????
????word:?詞語
????return:?詞向量
????“““
????
????if?word?in?self.model:
??????return?self.model[word]
????return?None
????
??def?word_similarity（self?word1?word2）:
????“““
????計算詞語相似度
????
????word1:?詞語1
????word2:?詞語2
????return:?詞語1與詞語2的相似度
????“““
????
????if?word1?not?in?self.model?or?word2?not?i