如何获取从模型文件加载的lightgbm Booster模型的参数?

3
有没有一种方法可以获取从模型文件加载的lightgbm Booster的参数字典?我已经优化了我的模型,然后使用这条线保存它:model.save_model('lgb-opt.txt'),然后使用这个加载:model = lgb.Booster(model_file='lgb-opt.txt)。现在,如果我尝试获取model.params,它会返回一个空字典。但是,如果我手动设置参数并重新训练,model.params会返回正确的参数字典。所以,设置是否在保存模型时丢失,或者仍然有一种方法可以获得它?再次优化只为获取参数集将非常耗时。谢谢。
1个回答

1

由于你的标签是lightgbtm,我认为lgb.Booster应该是"lightgbm.Booster"?

由于这个类有不同的参数,你应该明确地定义你使用的是哪个。尝试:

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

也许这会有所帮助。


Iwan Sivoronov,对不起,我漏掉了model_file这部分。你提到的那行代码是我实际使用的。 - undefined

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接