export_graphviz extratreesclassifier python可视化

Question

有没有办法使用export_graphviz代表extratreesclasiffier的最终树？如果没有，它怎么能在视觉上表现出来？

Answer 1

我认为无法从 sklearn.ensemble.ExtraTreesClassifier 中可视化特定树。

但你可以用 sklearn.tree.ExtraTreeClassifier 中的一棵树来形象化它。

from sklearn import tree
...

#Fit an Extra Tree model to the data
model = tree.ExtraTreeClassifier()
model.fit(X_train, y_train)

#Use http://webgraphviz.com to visualize the graph of this file
with open("tree_classifier.txt", "w") as f:
    f = tree.export_graphviz(model, out_file=f)

这里的决策树也可以使用Extra树的示例： http://dataaspirant.com/2017/04/21/visualize-decision-tree-python-graphviz/