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

Question

我的问题如下：

当运行这段代码：

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

关于上面代码的第二行，我得到这个错误：

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

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

Answer 1

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

%sh 
apt-get -y install graphviz

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