如何在 python 中获取 xgb.train 的超参数

Question

xgb.train 是在 Python 中训练 xgboost 模型的低级别 API。

• 当我使用 XGBClassifier 时，它是一个包装器并调用 xgb.train 训练了一个模型，我可以打印 XGBClassifier 对象和 打印超参数。
• 当使用 xgb.train 我不知道如何检查参数 培训

代码：

bst = xgb.train(params, dtrain)
bst.params # does not work!

Answer 1

here 中提到的 save_config 方法可用于创建模型配置的字符串表示形式。这可以转换为字典：

import json

config = json.loads(bst.save_config())

结果嵌套的有点深，但是超参数是这样找到的：

config['learner']['gradient_booster']['updater']['grow_colmaker']['train_param']