是否有一些方法可以知道XGBoost树适用于哪个数据子样本？

Question

我正在做一些XGBoost练习，我想知道XGBRegressor的树是在哪个数据子集上拟合的。下面是我使用的参数列表：

params = {'learning_rate': 0.09, 
          'n_estimators': 5,
          'objective': 'reg:squarederror',
          'max_depth': 6,
          'subsample': 0.2 
         }

我正在使用Python 3.7和XGBoost库。有什么办法知道吗？我读过文件，但找不到有用的

Answer 1

你可以很容易地画出这棵树并检查它 你可以用它来做

export_graphiviz(uralgo,"filename",feature_names,rounded=True)