Визуализируйте дерево решений в колаборатории - PullRequest
Новая
       41

Визуализируйте дерево решений в колаборатории

0 голосов
/ 05 ноября 2018

Как лучше всего визуализировать дерево решений с помощью Google Colab? Визуализации из 'dtreeviz' (например, Github ) действительно аккуратны, но при запуске чего-то вроде 

!pip install dtreeviz

и 

from sklearn.datasets import *
from sklearn import tree
from dtreeviz.trees import *

с последующим 

classifier = tree.DecisionTreeClassifier(max_depth=4)
cancer = load_breast_cancer()
classifier.fit(cancer.data, cancer.target)
viz = dtreeviz(classifier,
              cancer.data,
              cancer.target,
              target_name='cancer',
              feature_names=cancer.feature_names, 
              class_names=["malignant", "benign"],
              fancy=False )  

viz.view()

Я получаю 

ExecutableNotFound: failed to execute ['dot', '-Tsvg', '-o', '/tmp/DTreeViz_62.svg', '/tmp/DTreeViz_62'], make sure the Graphviz executables are on your systems' PATH

Кто мог бы чем-то заняться, какой Colab работает через мой g-drive?

Любая помощь приветствуется!

1 Ответ

0 голосов
/ 20 июня 2019

Краткий ответ

  • Убедитесь, что graphviz установлен через !apt-get install graphviz
  • Вы можете получить созданный SVG через viz.svg()

  • Оберните вывод в IPython HTML и затем наберите display, чтобы показать его в вашей записной книжке 

    from IPython.core.display import display, HTML
display(HTML(viz.svg()))

Более длинный ответ

  • dtreeviz view() создает файл SVG в вашем временном каталоге
  • Этот файл передается в библиотеку graphviz, которая открывает его в зависимости от вашей ОС
  • Google colab распознается как linux и пытается открыть файл SVG через приложение для просмотра по умолчанию
  • Последний шаг ведет в никуда, если вы не запускаете ноутбук локально (возможно, на сервере Google теперь есть пара открытых изображений SVG или несколько сообщений об ошибках)
  • Код из краткого ответа просто получает код SVG без его сохранения и затем отображает его в блокноте
...