python - 如何保存到磁盘/导出在 python 中训练的 lightgbm LGBMRegressor 模型？

Question

您好，我无法找到一种方法将 lightgbm.LGBMRegressor 模型保存到文件中供以后重复使用。

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')

如果有帮助请告诉我