Sklearn GradientBoostingClassifierを視覚化するには？

Question

私はgradient boost classifierを訓練しました。hereと表示されたgraphviz_exporterツールを使用して視覚化したいと思います。

私はそれをしようとすると、私が得る：

AttributeError: 'GradientBoostingClassifier' object has no attribute 'tree_' 

graphviz_exporterがdecision treesのためのものですので、これはですが、私は、勾配ブースト分類器は、基本的な決定木を持っている必要がありますので、それを可視化する方法は、まだだと思います。

誰かがそれを行う方法を知っていますか？

Answer 1

アトリビュートエスティメータには、基本的なデシジョンツリーが含まれています。次のコードは、訓練されたGradientBoostingClassifierのツリーの1つを表示します。

clf = GradientBoostingClassifier(
    n_estimators=200, 
    learning_rate=1.0, 
    max_depth=3, 
    random_state=42 
) 
clf = clf.fit(X[:600], Y[:600]) 


# Get the tree number 42 
sub_tree_42 = clf.estimators_[42, 0] 

dot_data = tree.export_graphviz(
    sub_tree_42, 
    out_file=None, filled=True, 
    rounded=True, 
    special_characters=True, 
    proportion=True, 
) 
graph = pydotplus.graph_from_dot_data(dot_data) 
Image(graph.create_png())