可视化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')

但这会返回：

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')