我试图在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之后,我仍然存在这个问题。
更新12/13/2020:我能够解决这个问题。这就是我所做的:
在我的其他导入语句块下面,我放了一个这样的块:
import os
os.environ['PATH'] = os.environ['PATH']+';'+os.environ['CONDA_PREFIX']+r"Librarybingraphviz"
我也遇到了同样的问题。这需要设置一些环境变量。所以我建议运行以下命令
conda install python-graphviz
而不是
pip install graphviz
以获得这些绑定,这些绑定也可以与conda的Graphviz包一起使用。