Ошибка при попытке преобразовать график для дерева решений из .dot в .png

#python #databricks #graphviz #decision-tree

Вопрос:

Моя проблема заключается в следующем:

При выполнении этого фрагмента кода:

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

Я получаю эту ошибку, касающуюся второй строки кода выше:

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

Я видел несколько сообщений, касающихся подобных вопросов, но ни в одном из них конкретно не говорилось о базах данных. Я понятия не имею, как найти, где установлен пакет, и добавить его в переменную среды PATH

Ответ №1:

dot Программа-это часть graphviz пакета, которую можно установить на компьютер с помощью диспетчера пакетов. Просто выполни:

 %sh 
apt-get -y install graphviz
 

Это %sh позволяет выполнять команды оболочки из ноутбука. Примите во внимание, что они выполняются только на узле драйвера, а не на всех узлах кластера. Если вам нужно будет установить что-то подобное на всех узлах или сделать это автоматически при запуске кластера, вы можете использовать сценарии инициализации кластера.