如何提取梯度老板分类器的决策规则



我想提取和可视化scikit-learn GradientBoostingClassifier的决策规则。如果它只是一个决策树分类器,我会使用tree.export_graphviz,但GradientBoostingClassifier是树的集合。我不知道我将如何在他们身上使用export_graphviz

如果有人知道一种方法,那将非常有帮助。

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

最新更新