python - 获取 GBDT 模型树信息

Question

我正在使用GBDT来自sklearn learn，我想知道是否有什么方法可以获得最终训练好的GBDT树信息？我的理解是，如果我设置最大 500 棵树和每棵树最大 10 深度，这是一个上限，我想得到实际使用的树数量，以及每棵树的实际深度。

Answer 1

您链接到的文档页面列出了以下属性:

estimators_ : ndarray of DecisionTreeRegressor,shape (n_estimators, loss_.K)

    The collection of fitted sub-estimators. loss_.K is 1 for binary classification, otherwise n_classes.

因此，您应该能够按照树添加到模型的顺序获取各个树。

附加注释:模型中使用的实际树数将等于参数n_estimators，除非使用提前停止，否则它可能会更少，并且存储在以下属性中:

n_estimators_ : int

    The number of estimators as selected by early stopping (if n_iter_no_change is specified). Otherwise it is set to n_estimators

除非每个叶子/分割没有足够的样本以及其他参数限制，否则深度将最大化。