如何在 Python 中生成 Word2vec 向量？

Question

我正在尝试生成 Word2vec 向量。

我有 pandas 个数据框。

我把它变成了代币

df["token"]

使用了 gensim.models

中的 Word2vec

model = w2v.Word2Vec(
sentences=df["token"],
seed=seed,
workers=num_workers,
size=num_features,
min_count=min_word_count,
window=context_size,
sample=downsampling
)

现在如何转换我的数据框 df？

这相当于

model.transform(df)

Answer 1

如果你的数据框只由单词组成，你可以

df['new_column'] = model[df['words']]

model['word'] 或 model[list()] 都为您提供 word 或 list

的矢量表示