【发布时间】:2019-02-26 06:26:08
【问题描述】:
处理工资数据集。一切正常,除非我在 Spyder 3.3.2 中使用 python-graphviz 模块来显示决策树图,它只是在控制台窗口中显示一个图像图标。同样的事情也适用于其他系统。我在这里缺少什么?
输出图片在这里Console Output
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)
【问题讨论】:
-
这在 Jupyter 笔记本中是否适合您?
-
是的,在 Jupyter notebook 中工作。
-
好的,我会在下面添加我的答案。
标签: python-3.x spyder graphviz