如何可視化Sklearn GradientBoostingClassifier？

Question

我已經訓練了gradient boost classifier，我想使用graphviz_exporter工具將其顯示爲here。

當我嘗試它，我得到：

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

這是因爲graphviz_exporter是爲decision trees，但我想還是有辦法對其進行可視化，因爲梯度升壓分類必須有一個基本的決策樹。

有人知道該怎麼做嗎？

Answer 1

屬性估計器包含基礎決策樹。以下代碼顯示受過訓練的GradientBoostingClassifier的其中一棵樹。

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