首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >在word2vec中可以使用gensim deeplearning4j.word2vec模型吗?

在word2vec中可以使用gensim deeplearning4j.word2vec模型吗?
EN

Stack Overflow用户
提问于 2017-04-26 11:37:02
回答 1查看 3.2K关注 0票数 6

我是deeplearning4j的新手,我想用单词向量作为分类器的输入来制作句子分类器。我以前使用python,其中向量模型是使用gensim生成的,我想使用这个新的分类器使用这个模型。在word2vec中可以使用gensim的deeplearning4j.word2vec模型吗?我是如何做到的?

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2018-07-11 20:02:42

是的,这是可能的,因为Word2Vec实现定义了一个标准来构造它的模型。

为此,请执行以下操作:

  1. 使用gensim,保存与Word2Vec实现兼容的模型: w2v_model.wv.save_word2vec_format("path/to/w2v_model.bin",binary=True)
  2. 在DL4J中,加载相同的预训练模型: WordVectorSerializer.readWord2VecModel("path/to/w2v_model.bin");Word2Vec w2vModel =

实际上,您可以在这两种代码中测试模型,并且您应该看到相同的结果,例如:

与天才:

代码语言:javascript
复制
print(w2v_model.most_similar("love"))
print(w2v_model.n_similarity(["man"], ["king"]))

对于DL4J:

代码语言:javascript
复制
System.out.println(w2vModel.wordsNearest("love", 10));
System.out.println(w2vModel.similarity("man", "king"));
票数 10
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/43633092

复制
相关文章

相似问题

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