1. 程式人生 > >決策樹的實現和排序重要特征

決策樹的實現和排序重要特征

https 字符 learn htm ocs red class ren clas

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]])
  • 訓練的決策樹模型為tree,返回決策樹訓練的前五個重要特征的list
  • 返回排序的索引用argsort函數,函數用法見https://docs.scipy.org/doc/numpy/reference/generated/numpy.argsort.html#numpy.argsort
  • 取倒數5個字符用了[::-1][0:5]這樣的索引

決策樹的實現和排序重要特征