使用gensim word2vec訓練英文預料—最簡潔版

直接上程式碼：

# -*- coding: utf-8 -*-
import gensim.models.word2vec as word2vec
import gensim
import logging

# word2vec Text8 的訓練
def train_save_model():
    # logging.basicConfig(format='%(asctime)s : %(levelname)s :%(message)',level=logging.INFO)
    # 載入預料
    sentences = word2vec.Text8Corpus('text8'
 
)
    model = word2vec.Word2Vec(sentences, size=200)
    model.save('text.model')
# 載入模型
def load_model():
    model = word2vec.Word2Vec.load('text.model')
    # simi = model.similar_by_vector('women', 'men')
    # print(simi)
    print(model.most_similar('man'))
    print(model['red'])

# 執行程式碼
load_model()