gpt4 book ai didi

python - 从 selectKbest 获取特征名称

转载 作者:行者123 更新时间:2023-12-01 08:22:10 26 4
gpt4 key购买 nike

我使用 Scikit learn selectKbest 选择最佳特征,从 900 个特征中选择大约 500 个。如下,其中 d 是所有特征的数据框。

from sklearn.feature_selection import SelectKBest, chi2, f_classif
X_new = SelectKBest(chi2, k=491).fit_transform(d, label_vs)

当我现在打印X_new时,它只给我数字,但我需要所选功能的名称以便稍后使用它们。

我尝试了诸如X_new.dtype.names之类的东西,但我没有得到任何结果,我尝试将X_new转换为数据框,但我得到的唯一列名称是

1, 2, 3, 4... 

有没有办法知道所选功能的名称是什么?

最佳答案

以下是使用 get_support() 的方法:

chY = SelectKBest(chi2, k=491)
X_new = chY.fit_transform(d, label_vs)
column_names = [column[0] for column in zip(d.columns,chY.get_support()) if column[1]]

从@AI_Learning的答案中,您可以通过以下方式获取列名称:

column_names = d.columns[chY.get_support()]

关于python - 从 selectKbest 获取特征名称,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54560611/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com