你好,我找不到一种方法来将lightgbm.LGBMRegressor
模型保存到文件中以便稍后重用。
回答:
对于Python 3.7和lightgbm==2.3.1
,我发现之前的答案不足以正确保存和加载模型。以下方法有效:
lgbr = lightgbm.LGBMRegressor(num_estimators = 200, max_depth=5)lgbr.fit(train[num_columns], train["prep_time_seconds"])preds = lgbr.predict(predict[num_columns])lgbr.booster_.save_model('lgbr_base.txt')
最后,我们可以通过以下方式验证这个方法是否有效:
model = lightgbm.Booster(model_file='lgbr_base.txt')model.predict(predict[num_columns])
如果没有上述方法,我会得到错误:AttributeError: 'LGBMRegressor' object has no attribute 'save_model'