from sklearn.feature_extraction.text import TfidfVectorizer vectorizer = TfidfVectorizer(max_features=512) def embed(text): vec = vectorizer.fit_transform([text]) return vec.toarray()[0].tolist()