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

Question

我正在使用 python 创建决策树分类器。我想查看我在 classier 中使用过的功能 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')

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