如何将在Python中训练的lightgbm LGBMRegressor模型保存到磁盘/导出?

你好,我找不到一种方法来将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'

Related Posts

关于k折交叉验证的直观问题

我在使用交叉验证检查预测能力时遇到了一些直观问题,我认…

调整numpy数组大小以使用sklearn的train_test_split函数?

我正在尝试使用sklearn中的test_train_…

如何转换二维张量和索引张量以便用于torch.nn.utils.rnn.pack_sequence

我有一组序列,格式如下: sequences = to…

模型预测值的含义是什么?

我在网上找到一个数字识别器的CNN模型并进行了训练,当…

锯齿张量作为LSTM的输入

了解锯齿张量以及如何在TensorFlow中使用它们。…

如何告诉SciKit的LinearRegression模型预测值不能小于零?

我有以下代码,尝试根据非价格基础特征来估值股票。 pr…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注