在 Word2Vec 向量空间中以特定方向移动

Question

我使用 gensim 训练了一个 Word2Vec 模型，并且想查询附近的术语。但不仅仅是获取所有方向上最接近的单词：

model = models.Word2Vec.load('MyModel')        # load up my trained model
nearest = model.most_similar(['mushroom'])     # nearby words all around

我想在向量空间内沿特定方向和距离移动并检索最近的单词，本质上是：

nearest = nearest_by_vector(word, direction_vector)

我的向量数学很糟糕（即不存在），尤其是我的模型中有这么多维度。

Answer 1

我认为您的问题是找到距离最近的向量（欧几里德距离），而不是角度最近的向量。这本质上是Nearest Neighbor Search的问题。最坏情况/蛮力搜索算法将是线性搜索。但是，您可以在 Google 上搜索各种高级库或数据结构，这些库或数据结构将以更优化和更具成本效益的方式找到这些最近的邻居。