File size: 241 Bytes
5c5b473
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
from sklearn.feature_extraction.text import TfidfVectorizer

vectorizer = TfidfVectorizer(max_features=5000)

def fit_vectorizer(texts):
    return vectorizer.fit(texts)

def transform(texts):
    return vectorizer.transform(texts).toarray()