我想提取并可视化scikit-learn GradientBoostingClassifier
的决策规则。如果它只是一个DecisionTreeClassifier,我会使用tree.export_graphviz
,但GradientBoostingClassifier
是一组树。我不知道我如何使用export_graphviz
。
如果有人知道如何做到这一点,那将非常有帮助。
答案 0 :(得分:2)
有人解决了这个问题。 “您可以使用clf.estimators_来检索与每个提升阶段相对应的DecisionTreeRegressor对象,并调用export_graphviz来生成树形可视化。 http://scikit-learn.org/stable/modules/generated/sklearn.ensemble.GradientBoostingClassifier.html“