可视化随机森林回归树

Question

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

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

我正在尝试这个:

import pydotplus
dot_data=r'F:SheyenneRandom_Forestmy_tree0.dot'
graph = pydotplus.graph_from_dot_data(dot_data) 
graph.write_pdf(r'F:SheyenneRandom_Forestmy_tree0.pdf')

但是返回:

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

Answer 1

看起来您正在尝试加载文件。试试这个:

import pydotplus
dot_file=r'F:SheyenneRandom_Forestmy_tree0.dot'
graph = pydotplus.graph_from_dot_file(dot_file) 
graph.write_pdf(r'F:SheyenneRandom_Forestmy_tree0.pdf')