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

解决方案

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

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐