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

Question

我正在使用python创建一个决策树分类器。我想要查看我在分类器中使用的功能名称。以下是代码

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')

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