首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >如何加快Gensim Word2vec模型的加载时间?

如何加快Gensim Word2vec模型的加载时间?
EN

Stack Overflow用户
提问于 2017-03-24 04:30:59
回答 3查看 24.5K关注 0票数 24

我正在构建一个聊天机器人,所以我需要使用Word2Vec向量化用户的输入。

我使用的是Google提供的一个有300万字的预训练模型(GoogleNews-vectors negative300)。

因此,我使用Gensim加载模型:

代码语言:javascript
复制
import gensim
model = gensim.models.KeyedVectors.load_word2vec_format('GoogleNews-vectors-negative300.bin', binary=True)

问题是加载模型大约需要2分钟。我不能让用户等那么久。

那么,我可以做些什么来加快加载时间呢?

我考虑将300万个单词及其对应的向量放入一个MongoDB数据库中。这当然会加快速度,但直觉告诉我这不是一个好主意。

EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/42986405

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档