python——word2vec

这篇博客介绍了如何利用gensim库训练word2vec模型,以yelp和amazon电子评论为语料,进行了预处理和词向量的训练。讨论了min_count和size等关键参数对模型的影响。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

使用gensim提供的模型word2vec训练从网上获得的语料。gensim是Radim Řehůřek博士提供的库。

比较简单的参考资料:getting-started-with-word2vec-and-glove-in-python

更进一步的word2vec的使用可以参考这篇文章:

Word2vec Tutorial


1、处理原始语料,得到word2vec需要的输入格式

这次实验使用的语料是yelp(共2225213条)和amazon electronic(共计1689188条)上的评论语料,都是json格式的。

首先清楚word2vec模型的输入是什么样子的。可以是内建的列表,其中每个元素表示一句话,每个元素又是一个列表组成,比如

sentences=[['i', 'am', 'wei'], ['hello','everyone']]
model = gensim.models.Word2Vec(sentences, min_count=1)

当然这样只能训练出来的词向量没有任何意思。也不可能把所有的语料全部存在一个list中,全部加载进内存。word2vec还提供了一个通过读取文件的方式获得输入:

word2vec(filenam
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值