是否有一些方法可以知道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)