当前位置:   article > 正文

LightGBM模型保存save_model后加载_lightgbm保存模型

lightgbm保存模型

LightGBM模型的加载通常会使用外置的joblib接口来实现,但有时候我们也能看到使用其自带接口save_model来实现。但网上对于这一接口下保存的模型该如何读取,并没有相关详尽的介绍。本文将给出model.save_model('model.txt')保存后后该如何读取

解决方案

 使用model.save_model('model.txt')对模型进行保存后,使用model=lgb.Booster(model_file='model.txt')读取即可。注意,是直接用Booster,不是当初模型保存时对应的LGBMRegressorLGBMClassifier*

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/菜鸟追梦旅行/article/detail/258025
推荐阅读
相关标签
  

闽ICP备14008679号