特征选择python
时间: 2023-09-11 14:05:28 浏览: 144
自动特征选择Python代码实现,ipynb格式
5星 · 资源好评率100%
在Python中,有多种方法可以进行特征选择。以下是一些常用的技术和库:
1. 方差选择:使用VarianceThreshold类来删除方差低于阈值的特征。
```python
from sklearn.feature_selection import VarianceThreshold
selector = VarianceThreshold(threshold=0.1)
X_selected = selector.fit_transform(X)
```
2. 相关系数选择:使用Pearson相关系数或其他统计指标来选择与目标变量相关性较高的特征。
```python
import pandas as pd
corr_matrix = df.corr()
# 选择与目标变量相关性大于阈值的特征
selected_features = corr_matrix[abs(corr_matrix['target']) > 0.5].index
```
3. 递归特征消除(Recursive Feature Elimination, RFE):使用递归方式逐步删除对模型预测性能贡献较小的特征。
```python
from sklearn.feature_selection import RFE
from sklearn.linear_model import LinearRegression
estimator = LinearRegression()
selector = RFE(estimator, n_features_to_select=5)
X_selected = selector.fit_transform(X, y)
```
4. 基于树模型的特征选择:使用决策树、随机森林等模型,根据特征的重要性进行选择。
```python
from sklearn.ensemble import RandomForestClassifier
model = RandomForestClassifier()
model.fit(X, y)
# 根据特征的重要性排序
importance = model.feature_importances_
sorted_features = [x for _, x in sorted(zip(importance, X.columns), reverse=True)]
```
这些是特征选择的一些常见方法和库,根据具体的应用场景和数据集,选择适合的方法进行特征选择。
阅读全文