首页 > 解决方案 > 如何获取从模型文件加载的 lightgbm Booster 模型的参数?

问题描述

有没有办法获取从模型文件加载的 lightgbm Booster 的参数字典?我已经优化了我的模型,然后使用这一行保存它,然后用这一行model.save_model('lgb-opt.txt')加载model = lgb.Booster(model_file='lgb-opt.txt)。现在,如果我尝试获取model.params,它会返回一个空字典。但是,如果我手动设置参数并再次训练,model.params则会返回正确的参数字典。那么,在保存模型的过程中,套装会丢失,还是有办法获得呢?仅针对参数集再次优化将非常耗时。

谢谢你。

标签: pythonlightgbm

解决方案


由于您的标签是 lightgbtm,因此假设 lgb.Booster 应该是“lightgbm.Booster”?

由于这个类有不同的参数,你应该专门定义,你正在使用它们中的哪一个。尝试:

model = lgb.Booster(model_file='lgb-opt.txt')

也许这会有所帮助。


推荐阅读