Работа над набором данных о зарплате.Все работает отлично, за исключением случаев, когда я использую модуль python-graphviz
в Spyder 3.3.2 для отображения графика дерева решений, он просто показывает значок изображения в окне консоли.То же самое работает и в других системах.Что мне здесь не хватает?Выходное изображение здесь. Консольный вывод
from sklearn.tree import DecisionTreeClassifier
dtf = DecisionTreeClassifier()
dtf.fit(X_train, y_train)
from sklearn.tree import export_graphviz
export_graphviz(dtf, out_file="tree.dot", class_names=["Less than 50k",
"More than 50k"])
import graphviz
with open("tree.dot") as f:
dot_graph = f.read()
graphviz.Source(dot_graph)