利用python对Boston数据集进行特征选择的代码
时间: 2024-05-09 07:17:00 浏览: 106
Python进行特征提取的示例代码
以下是利用Python对Boston数据集进行特征选择的代码示例:
```python
import pandas as pd
from sklearn.datasets import load_boston
from sklearn.feature_selection import SelectKBest, f_regression
# 加载数据集
boston = load_boston()
data = pd.DataFrame(boston.data, columns=boston.feature_names)
target = boston.target
# 特征选择
selector = SelectKBest(f_regression, k=5)
selector.fit(data, target)
selected_features = data.columns[selector.get_support()]
# 输出选出来的特征
print(selected_features)
```
代码中使用了pandas库加载数据集,sklearn.feature_selection库进行特征选择,选择F值排名前5名的特征。最后输出选出来的特征名称。
阅读全文