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