【发布时间】:2018-12-02 04:03:49
【问题描述】:
我查看了几个示例并遵循了它们,但无法打印树形图。
R_forest = RandomForestRegressor(bootstrap=False,
max_depth=30,
max_features ='sqrt',
min_samples_leaf= 4,
min_samples_split=2,
n_estimators = 600)
model=R_forest.fit(X_train,y_train)
from sklearn.datasets import *
from sklearn import tree
from sklearn.tree import export_graphviz
import graphviz
import pydot
tree = R_forest.estimators_[5]
# Export the image to a dot file
export_graphviz(tree, out_file = 'tree.dot', feature_names = X_train.columns,
rounded = True, precision = 1)
# Use dot file to create a graph
(graph, ) = pydot.graph_from_dot_file('tree.dot')
# Write graph to a png file
graph.write_png('tree.png')
我收到此错误:
FileNotFoundError: [WinError 2] "dot.exe" not found in path.
我遵循了这个解决方案,但仍然遇到同样的错误。
我的系统截图。
任何帮助或建议表示赞赏
【问题讨论】:
-
哪一行代码给了你错误?你能生成“tree.dot”文件吗?
-
是的,我能够生成“tree.dot”文件。感谢您的回答,我将确认为正确答案。
标签: graphviz random-forest pydot