尝试将决策树的图从 .dot 转换为 .png 时出错

Question

我的问题如下：

运行这段代码时：

(graph,) = graph_from_dot_file('tree.dot')
graph.write_png('/mnt/tree1.png')

我收到此错误，关于上面代码的第二行：

FileNotFoundError: [Errno 2] "dot" not found in path.

我看过一些关于类似问题的帖子，但没有一个专门讨论数据块。我不知道如何找到软件包的安装位置并将其添加到我的 PATH 环境变量中

Answer 1

dot 程序是graphviz 包的一部分，可以使用包管理器安装到机器上。只需执行：

%sh 
apt-get -y install graphviz

%sh 允许您从笔记本执行 shell 命令。考虑到它们仅在驱动节点上执行，而不是在集群的所有节点上执行。如果你需要在所有节点上安装类似的东西，或者在集群启动时自动安装，你可以使用cluster init scripts。