python - 什么相当于 Python 中的 rpart.plot？我想可视化我的随机森林的结果

Question

在 [R] 中，您可以像这样可视化随机森林的结果(图片无耻地从互联网上盗用)。 Python中的等价物是什么？我可以使用 feature_importances_ 获得我的 sklearn 随机森林分类的​​结果，但我想知道他们将结果发送到哪个方向。我意识到在森林深处我无法检查每个分支，但也许它可以衡量概率？谢谢。

Answer 1

使用tree.export_graphviz的参数，看这个例子和图片结果:

tree.export_graphviz(clf, out_file=out, feature_names=['alcohol', 'income'],class_names=['0','1'], 
                     filled=True, rounded=True, special_characters=True)

graph=pydotplus.graph_from_dot_data(out.getvalue())
Image(graph.create_png())

图像结果

参见 jupyter notebook on github和 project details .