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. There is examples方法来绘制您自己的数据结构。