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


def do_tfidf(token):
    tfidf = TfidfVectorizer(max_df=0.05, min_df=0.002)
    words = tfidf.fit_transform(token)
    sentence = " ".join(tfidf.get_feature_names())
    return sentence