可视化 RandomForestRegression 树

Question

我在 sklearn 中有一个运行 RandomForestRegression 模型，并将我的决策树 (n_estimators=50) 的输出保存到 50 个 .dot 文件中。

现在我想保存它们，这样我就可以将它们视为真正的树。

我正在尝试这个：

import pydotplus

dot_data=r'F:\Sheyenne\Random_Forest\my_tree0.dot'

graph = pydotplus.graph_from_dot_data(dot_data) 

graph.write_pdf(r'F:\Sheyenne\Random_Forest\my_tree0.pdf')

但是这个returns:

AttributeError: 'NoneType' object has no attribute 'write_pdf'

Answer 1

您似乎正在尝试加载文件。试试这个：

import pydotplus

dot_file=r'F:\Sheyenne\Random_Forest\my_tree0.dot'

graph = pydotplus.graph_from_dot_file(dot_file) 

graph.write_pdf(r'F:\Sheyenne\Random_Forest\my_tree0.pdf')

可视化 RandomForestRegression 树

Visualizing RandomForestRegression trees

pydot

scikit-learn