Python进行特征选择的代码
时间: 2023-11-19 10:27:23 浏览: 130
可以通过使用scikit-learn库中的特征选择函数或者通过手动实现选择算法来进行特征选择,以下是一个示例代码:
from sklearn.feature_selection import SelectKBest, chi2
# 假设X为特征矩阵,y为目标变量
X_new = SelectKBest(chi2, k=5).fit_transform(X, y)
这里的代码使用SelectKBest函数,选择了5个最佳特征,并使用卡方检验来评估特征对目标变量的相关性。
阅读全文