16.【进阶】特征提升之特征筛选----feature_selection

最新推荐文章于 2024-04-21 13:19:23 发布

jho9o5

最新推荐文章于 2024-04-21 13:19:23 发布

阅读量1.3k

点赞数 1

分类专栏：机器学习实战文章标签：机器学习特征筛选泰坦尼克号

本文为博主原创文章，未经博主允许不得转载。

本文链接：https://blog.csdn.net/jh1137921986/article/details/79822512

版权

本文深入探讨了在机器学习项目中如何进行特征筛选，以提升模型性能。以泰坦尼克号数据集为例，详细阐述了特征筛选的重要性和具体方法，包括基于统计的方法和基于模型的方法，旨在帮助读者掌握有效的特征选择技巧。

摘要由CSDN通过智能技术生成

#-*- coding:utf-8 -*-

#特征筛选的目标：一句话来说就是去冗余特征；与PCA不同，不会修改特征值，而是寻找那些对模型性能提升较大的少量特征

#使用Titanic数据集，通过特征筛选的方法一步步提升决策树的预测性能

import pandas as pd
titanic = pd.read_csv('http://biostat.mc.vanderbilt.edu/wiki/pub/Main/DataSets/titanic.txt')

#分离数据特征与预测目标
y = titanic['survived']
X = titanic.drop(['row.names','name','survived'],axis=1)

#填充缺失数据
X['age'].fillna(X['age'].mean(),inplace=True)
X.fillna('UNKNOWN',inplace=True)

#分割数据
from sklearn.cross_validation import train_test_split
X_train,X_test,y_train,y_test = train_test_split(X,y,test_size=0.25,random_state=33)

#类别型特征向量化
from sklearn.feature_extraction import DictVect