决策树的实现和排序重要特征

时间：2017-09-07 20:23:30 阅读：509 评论：0 收藏：0 [点我收藏+]

标签：from nump 五个 doc list sklearn col rand 函数

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]这样的索引

决策树的实现和排序重要特征

标签：from nump 五个 doc list sklearn col rand 函数

原文地址：http://www.cnblogs.com/sss423/p/7491537.html

踩

(0)

评论一句话评论（0）