python - 从 Tfidf Vectorizer 中仅选择前 n 个特征

Question

var_vector = TfidfVectorizer()
train_var = var_vector.fit_transform(t_df['var'])

top_100 = np.array(var_vector.get_feature_names())
tfidf_100 = np.argsort(var_vector.idf_)[::-1]

n = 100
top_n = top_100[tfidf_100][:n]

从 tfidf Vectorizer 中选择前 100 个单词后，如何将维度更新为 100？

Answer 1

将max_features参数设置为100。请参阅此处的文档