python - 有没有办法返回由 Pycaret 调整的超参数？

Question

Pycaret 自动搜索最佳参数。例如，下面的代码会将 5 个自动调整的模型分配给“tuned_top5”。

from pycaret.classification import *
setup(data=train, train_size=.9, target='my_target_feature')
tuned_top5 = [tune_model(model) for model in top5]

然而，这对我来说还不够。我想知道超参数的确切名称和值。例如，如果此代码将 max_depth 调整为 9，我希望打印“max_depth=9”或类似的结果。

有没有办法做到这一点？

Answer 1

您可以从模型中提取参数名称并直接打印出来：

print(tuned_top5.get_all_params())

如果您只对单个参数或参数的简短列表感兴趣，则将它们单独拉出来打印。

params = tuned_top5.get_all_params()
print("max depth = ', params['max_depth'])