python - 在 Python 中从 Doc2Vec 中提取特征

Question

对于一个小项目，我需要提取从 gensim 中的 Doc2Vec 对象获得的特征。

我用过对vector = model.infer_vector(words)吗？

Answer 1

是的，如果words是一个字符串列表，预处理/标记化的方式与model在训练期间将训练数据馈送到的方式相同。