從Python中的Doc2Vec中提取特徵

Question

對於一個小型項目，我需要從gensim中的Doc2Vec對象中提取特徵。

我已使用vector = model.infer_vector(words)是對的嗎？

Answer 1

是的，如果words是一個字串列表，則預處理/標記化方式與培訓數據在訓練期間被送入model的方式相同。