Сохранение графика дерева решений Graphviz с использованием Scikit Learn DecisionTreeClassifier - PullRequest
0 голосов
/ 17 января 2019

Я могу сгенерировать график дерева решений с помощью graphviz, но когда я пытаюсь его сохранить (т.е. out_file)

Я получаю следующую ошибку:

CalledProcessError: Команда '[' dot ',' -Tsvg ']' вернула ненулевой выход статус 1.

Вот мой код:

# create plot for decision tree
dot_data = tree.export_graphviz(model, 
                                out_file='tree.dot', # this is what triggers the error
                                feature_names=X_test.columns,
                                class_names=['Active','Churned']) 

graph = graphviz.Source(dot_data) 

# view plot 
graph

Аргумент out_file - это тот, который вызывает ошибку. Я думаю, что это должно быть довольно простое решение, но я не нашел ответа на мои поиски.

Заранее спасибо.

1 Ответ

0 голосов
/ 19 марта 2019

Вы должны добавить параметр "out_file = None" в конце tree.export_graphviz () .

...