from gensim.models import word2vec
raw_sentence = ['我 今天 下午 打 篮球','我 今天 下午 打 足球','我 热爱 篮球 运动','他 是 足球 运动 ']
sentence = [i.split() for i in raw_sentence]
print(sentence)
model = word2vec.Word2Vec(sentence,min_count=1)
testword = '足球'
res = model.most_similar(testword)
print(res)
gensim训练word2vec
最新推荐文章于 2024-08-30 21:02:49 发布