使用CHI2测试仅选择20%最重要的功能时,如何获得功能名称



这是我在书中找到的代码狙击,"学习scikit-learn:

from sklearn import feature_selection
fs = feature_selection.SelectPercentile(feature_selection.chi2, percentile=20)
X_train_fs = fs.fit_transform(X_train, y_train)

我想要的是特征名称和相对重要性的数组。

我在堆栈中找到了此代码(在功能选择后显示功能名称),但不完全理解它,并且无法使其工作。

我将感谢有关如何获得,存储和打印功能的建议。感谢您的帮助。

这是edchum

的代码

.....将数据读为numpy数组

....获取功能名称并放入feature_names _

from sklearn import feature_selection
fs = feature_selection.SelectPercentile(feature_selection.chi2, percentile=20)
X_train_fs = fs.fit_transform(plants_X, plants_y)
print feature_names, 'n', fs.scores_

相关内容

  • 没有找到相关文章

最新更新