【发布时间】:2021-03-22 18:42:26
【问题描述】:
我尝试在 Jupyter Notebook(Python 版本 3.6.10)上可视化决策树,但无法实现。
from sklearn import tree
model = tree.DecisionTreeClassifier(criterion='entropy')
model=tree.DecisionTreeClassifier()
model.fit(trainData,trainLabel)
model.score(trainData,trainLabel)
predicted= model.predict(testData)
from io import StringIO
dot_data = StringIO()
tree.export_graphviz(model, out_file=dot_data)
graph = pydotplus.graph_from_dot_data(dot_data.getvalue())
Image(graph.create_png())
我不断收到“找不到 GraphViz 的可执行文件”。即使检查了其他线程同样的问题,添加环境变量,并重新启动 Jupyter,我仍然有这个问题。
2020 年 12 月 13 日更新:我能够解决此问题。这就是我所做的:
在我的其他导入语句的块下面,我放了一个这样的块:
import os
os.environ['PATH'] = os.environ['PATH']+';'+os.environ['CONDA_PREFIX']+r"\Library\bin\graphviz"
【问题讨论】:
-
你安装了graphviz / dot吗?您是否尝试从 jupyter 之外的命令提示符启动 dot?
-
是的,我确实安装了 graphviz。我尝试从常规命令提示符运行 dot.exe,但没有任何反应。
标签: python python-3.x graphviz decision-tree