sklearn.tree.export_graphviz可以重用于自己的数据结构吗？

Question

我一直在寻找在Python中为PNG绘制抽象语法树的好方法。结合使用networkx和matplotlib似乎可以很好地完成工作。

但是我只是注意到https://scikit-learn.org/stable/modules/generated/sklearn.tree.export_graphviz.html的表现要好得多！这在使用sklearn生成随机森林时适用；它是特定于最终决策树的功能。

是否可以为上述功能或背后的代码版本提供任意树，以获得高质量的渲染？

Answer 1

您可以使用简单的graphviz。 examples提供了绘制自己的数据结构的方法。