小编van*_*man的帖子

使用 chi2 测试进行具有连续特征的特征选择(Scikit Learn)

我试图从许多连续特征中预测二元(分类)目标,并希望在进入模型拟合之前缩小特征空间。我注意到 SKLearn 的 Feature Selection 包中的 SelectKBest 类在 Iris 数据集上有以下示例(它也从连续特征预测二进制目标):

from sklearn.datasets import load_iris
from sklearn.feature_selection import SelectKBest
from sklearn.feature_selection import chi2
iris = load_iris()
X, y = iris.data, iris.target
X.shape
(150, 4)
X_new = SelectKBest(chi2, k=2).fit_transform(X, y)
X_new.shape
(150,2)
Run Code Online (Sandbox Code Playgroud)

该示例使用 chi2 检验来确定模型中应使用哪些特征。然而,我的理解是 chi2 测试严格用于我们具有预测分类性能的分类特征的情况。我不认为 chi2 测试可以用于这样的场景。我的理解有误吗?chi2 检验能否用于检验分类变量是否依赖于连续变量?

python chi-squared feature-selection scikit-learn

5
推荐指数
1
解决办法
1931
查看次数