使用决策树进行特征选择

"""
使用决策树进行特征选择:
决策树也是常用的特征选取方法。使用决策树集合(如随机森林等)也可以计算每个特征的相对重要性。
这些重要性能够辅助进行特征选择。该方法主要使用信息增益率来进行特征选择。
"""
from sklearn import datasets
from sklearn import metrics
from sklearn.ensemble import ExtraTreesClassifier

dataset = datasets.load_iris()
model = ExtraTreesClassifier()
model.fit(dataset.data, dataset.target)
#打印出每个特征的重要性
print(model.feature_importances_)

猜你喜欢

转载自blog.csdn.net/QYHuiiQ/article/details/97615797