如何提取GradientBosstingClassifier的决策规则

时间:2017-05-15 07:07:46

标签: python scikit-learn

我想提取并可视化scikit-learn GradientBoostingClassifier的决策规则。如果它只是一个DecisionTreeClassifier,我会使用tree.export_graphviz,但GradientBoostingClassifier是一组树。我不知道我如何使用export_graphviz

如果有人知道如何做到这一点,那将非常有帮助。

1 个答案:

答案 0 :(得分:2)

有人解决了这个问题。 “您可以使用clf.estimators_来检索与每个提升阶段相对应的DecisionTreeRegressor对象,并调用export_graphviz来生成树形可视化。 http://scikit-learn.org/stable/modules/generated/sklearn.ensemble.GradientBoostingClassifier.html