from sklearn.tree import DecisionTreeClassifier from adspy_shared_utilities import plot_feature_importances tree=DecisionTreeClassifier(random_state=0)
tree.fit(X_train2,y_train2) list(X_mush.columns[tree.feature_importances_.argsort()[::-1][0:5]])
原文：http://www.cnblogs.com/sss423/p/7491537.html