python - 将 LGBMRegressor 模型从 python lightgbm 包保存到光盘

Question

嗨，我无法找到将 LGBM 模型保存并重用到文件的方法。我使用了 python 包 lightgbm 和 LGBMRegressor 模型。能否请你帮忙？文档似乎没有有用的信息。我在 Spyder 上使用 python 3.5

标签： pythonlightgbm

Answer 1

尝试：

my_model.booster_.save_model('mode.txt')
#load from model:

bst = lgb.Booster(model_file='mode.txt')

注意：API 状态

bst = lgb.train(…)
bst.save_model('model.txt', num_iteration=bst.best_iteration)

根据版本，上述方法之一有效。对于通用，您还可以使用 pickle 或类似的东西来冻结您的模型。

import joblib
# save model
joblib.dump(my_model, 'lgb.pkl')
# load model
gbm_pickle = joblib.load('lgb.pkl')

让我知道这是否有帮助