word2vec详细介绍可点击链接阅读,https://www.jianshu.com/p/471d9bfbd72f。
本文主要想简单通过python来实现word2vec的基本用法。
直接上代码:
一、训练语料。
本文语料是一百条有关交通的新闻。
# -*- coding: utf-8 -*-
"""
Created on Tue Apr 23 18:53:04 2019
@author: sun
"""
from gensim.models import Word2Vec
from gensim.models.word2vec import LineSentence
import logging
logging.basicConfig(format='%(asctime)s : %(levelname)s : %(message)s', level=logging.INFO)
# Word2Vec第一个参数代表要训练的语料
# sg=1 表示使用Skip-Gram模型进行训练