python - 获取每个文档的最高术语 - scikit tf-idf

Question

使用 scikit's tf-idf vectorizer 对多个文档进行矢量化后，有没有办法获得每个文档中最具“影响力”的术语？

不过，我只找到了获取整个语料库最具“影响力”术语的方法，而不是每个文档。

Answer 1

只需在 Ami 的最后两个步骤中添加另一种方法:

# Get a list of all the keywords by calling function
feature_names = np.array(count_vect.get_feature_names())
feature_names[X_train_tfidf.argmax(axis=1)]