python 特征选择特征工程

Posted 2021-05-09

#下面的树算法(Tree algorithms)计算特征的信息量：
#输出每个特征的重要程度：
#[0.13784722  0.15383598  0.25451389  0.17476852  0.02847222  0.12314815  0.12741402]

from sklearn import metrics
from sklearn.ensemble import ExtraTreesClassifier
model = ExtraTreesClassifier()
model.fit(X, y)
# display the relative importance of each attribute
print(model.feature_importances_)