热线电话:13121318867

登录
2018-11-29 阅读量: 828
基于树的特征选择

基于树的 estimators 可以用来计算特征的重要性,然后可以消除不相关的特征(当与 sklearn.feature_selection.SelectFromModel 等元转换器一同使用时):

>>>>>> from sklearn.ensemble import ExtraTreesClassifier 
>>> from sklearn.datasets import load_iris
>>> from sklearn.feature_selection import SelectFromModel
>>> iris = load_iris()
>>> X, y = iris.data, iris.target
>>> X.shape (150, 4)
>>> clf = ExtraTreesClassifier()
>>> clf = clf.fit(X, y)
>>> clf.feature_importances_ array([ 0.04..., 0.05..., 0.4..., 0.4...])
>>> model = SelectFromModel(clf, prefit=True)
>>> X_new = model.transform(X) >>> X_new.shape (150, 2)
0.0000
1
关注作者
收藏
评论(0)

发表评论

暂无数据
推荐帖子