python - 如何在 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

注意save_config 方法here可用于创建模型配置的字符串表示。这可以转换为字典:

import json

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

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

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