python - 使用 glove-python 模型获取词嵌入字典

Question

我使用 Maciejkula 的实现 (github repo) 在 python 中训练了一个 Glove 模型。对于下一步，我需要一个词到嵌入字典。但是，我似乎找不到一种简单的方法来从我训练的手套模型中提取这样的字典。

我可以通过访问 model.word_vectors 来提取嵌入，但这只会返回一个包含向量的数组，而不会映射到相应的词。还有包含词到索引对的 model.dictionary 属性。我认为这些索引可能对应于 model.word_vectors 数组中的嵌入索引，但我不确定这是正确的。

索引是否对应，或者是否有另一种简单的方法从 glove-python 模型中获取词到嵌入字典？

我意识到 Sanj我问了一个类似但更广泛的问题，但由于还没有回应，我想我会问这个更具体的问题。

Answer 1

您走在正确的轨道上。 NLP 解决方案通常避免在整个算法中保留单词，并使用索引方案，单词 -> idx，为了简单起见，该 idx 在整个算法中使用。

对于这个手套实现，model.dictionary 包含 word -> idx 而 model.word_vectors 包含 idx -> vectors。

例如要获取与单词“samsung”相对应的向量，您可以使用:

model.word_vectors[model.dictionary['samsung']]