File size: 207 Bytes
f283da3
d8c3a2c
f283da3
d8c3a2c
 
 
f283da3
 
 
1
2
3
4
5
6
7
8
9
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()