Scyrit-learn中GradientBoostingClassifier的叶子索引

时间:2015-03-18 20:56:30

标签: scikit-learn

有没有办法在GradientBoostingClassifier及其apply(X) method中为RandomForestClassifier获取{{1}}的叶子索引?

1 个答案:

答案 0 :(得分:2)

您可以在每个apply上调用estimator方法,我认为在当前版本中您必须调用gbrt.estimators[i].tree_.apply(X),这将在下一个版本中简化为gbrt.estimators[i].apply(X)发布iirc。