特征选择后显示特征名称 [英] show feature names after feature selection

问题描述

我需要为文本构建一个分类器，现在我使用 TfidfVectorizer 和 SelectKBest 来选择特征，如下所示:

I need to build a classifier for text, and now I'm using TfidfVectorizer and SelectKBest to selection the features, as following:

vectorizer = TfidfVectorizer(sublinear_tf = True, max_df = 0.5, stop_words = 'english',charset_error='strict')

X_train_features = vectorizer.fit_transform(data_train.data)
y_train_labels = data_train.target;

ch2 = SelectKBest(chi2, k = 1000)
X_train_features = ch2.fit_transform(X_train_features, y_train_labels)

我想在选择 k 个最佳特征后打印出所选特征名称(文本)，有什么办法可以做到?我只需要打印出选定的特征名称，也许我应该改用 CountVectorizer?

I want to print out selected features name(text) after select k best features, is there any way to do that? I just need to print out selected feature names, maybe I should use CountVectorizer instead?

推荐答案

以下应该有效:

np.asarray(vectorizer.get_feature_names())[ch2.get_support()]

这篇关于特征选择后显示特征名称的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！