如何寻找随机林树/决策树的特征？

Question

我正在使用python来创建决策树分类器。我想看看我从分类器中使用过的NAME功能。这是代码

In [3]: my_forest.feature_importances_
Out[3]: 
array([ 0.09546647,  0.210264  ,  0.31768782,  0.24191602,  0.05580487,
        0.04318897,  0.03567185])

我可以找到7个功能，但名称未显示。

Answer 1

使用以下方法将数据转换为字典：

dv = DictVectorizer()
newX = dv.fit_transform(all_data)

现在使用newX而不是X进行培训和测试。然后，你可以做

tree.export_graphviz(classifier_used, feature_names=dv.get_feature_names(), out_file='graph.dot')

这将帮助您可视化所有功能。