首页 > 解决方案 > 我们如何将我们自己的自定义嵌入与 WordMoverDistance 一起使用?

问题描述

要使用 WMD,我们需要一些词嵌入。对于此示例,使用了 Gensim ' word2vec-google-news-300 '提供的预训练嵌入。

下面是代码片段:

import gensim.downloader as api
model = api.load('word2vec-google-news-300')
distance = model.wmdistance(sent1, sent2)

如何使用我自己的自定义嵌入代替它,如何在模型中加载它?例如:嵌入有点像 - {text:1-D NumPy array}

标签: stanford-nlpgensimword2vecwmd

解决方案


推荐阅读