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

Question

export_graphviz(treeclf, out_file=' tree_泰坦尼克。点",feature_names = feature_cols)

在命令行，运行以下命令转换为PNG:

。Dot -o tree_titanic.png

我正在使用上面的代码，但是当我试图在终端中运行(点命令)时，它不起作用。我如何在jupyter中运行这个命令来可视化树?谢谢。

Answer 1

您可以在jupyter中通过在命令前加上!来运行终端命令。例如:

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

你可能需要在你的系统上安装Graphviz来使用点程序。

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