import gensim
sentences = [['你好', '呀',], ['今天', '天气',"很好"],['开心','高兴']]
# train word2vec on the this sentences
model = gensim.models.Word2Vec(sentences,min_count=1)
# get the detail
print(model.wv.vocab)
#get the words
words = model.wv.vocab
for word in words:
print(word)
# 获取model中的词 #在gensim 1.0.0 以前的版本可以使用:
model.vocab
#在 gensim 1.0以后的版本使用:
model.wv.vocab
第一个输出结果:
{'你好': <gensim.models.keyedvectors.Vocab object at 0x121f58a90>,
'呀': <gensim.models.keyedvectors.Vocab object at 0x1220ffa10>,
'今天': <gensim.models.keyedvectors.Vocab object at 0x12216ae90>,
'天气': <gensim.models.keyedvectors.Voca