how to extract decision rules of GradientBosstingClassifier

前端 未结 1 1721
长发绾君心
长发绾君心 2021-01-23 07:40

I want to extract and visualize the decision rules of a scikit-learn GradientBoostingClassifier. If it were just one DecisionTreeClassifier I\'d use tree.expo

相关标签:
1条回答
  • 2021-01-23 08:14

    Someone has solved the problem. "You can use clf.estimators_ to retrieve the DecisionTreeRegressor objects corresponding to each boosting stage and call export_graphviz to generate the tree visualization.. http://scikit-learn.org/stable/modules/generated/sklearn.ensemble.GradientBoostingClassifier.html"

    0 讨论(0)
提交回复
热议问题