如何从模型文件中加载lightgbm增压器模型的参数?

2024-09-28 05:26:05 发布

您现在位置:Python中文网/ 问答频道 /正文

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

多谢各位


Tags: 文件方法模型txt参数model字典save
1条回答
网友
1楼 · 发布于 2024-09-28 05:26:05

既然你的标签是lightgbtm,那么lgb.Booster应该是“lightgbm.Booster”吗

由于这个类有不同的参数,您应该具体定义,您使用的是哪一个参数。尝试:

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

也许这会有帮助

相关问题 更多 >

    热门问题