Extraire les fonctionnalités de Doc2Vec en Python

Question

Pour un petit projet, j'ai besoin d'extraire les fonctionnalités obtenues à partir de l'objet Doc2Vec en gensim. J'ai utilisé vector = model.infer_vector(words) est-ce correct?

Answer 1

Oui, si words est une liste de chaînes de mots, prétraitées/tokenisées de la même manière que les données d'entraînement ont été envoyées au model pendant la formation.