如何在Jupyter笔记本中打印决策树分类器？

Question

export_graphviz（treeclf，out_file ='tree_titanic.dot'，feature_names = feature_cols）

在命令行中，运行此命令以转换为PNG：

dot -Tpng tree_titanic.dot -o tree_titanic.png

我正在使用上面的代码，但是当我尝试在终端中运行（dot命令）时，它不起作用。如何在jupyter中运行此命令以可视化树？谢谢。

Answer 1

您可以通过在命令之前放置!来在jupyter中运行终端命令。例如：

! dot -Tpng tree_titanic.dot -o tree_titanic.png

您可能需要在系统上安装Graphviz才能使用点程序。

然后，您可以使用jupyter / ipython中的display子模块显示本地文件系统中的图像。