如何保存和加载已经训练好的LightGBM模型?
时间: 2024-09-13 20:17:22 浏览: 135
lightgbm模型保存为pmml文件 机器学习lgbm模型存为pmml文件 通过java调用jar
5星 · 资源好评率100%
在LightGBM中,你可以使用`dump_model()`方法将训练好的模型保存为文件,然后使用`load_model()`方法从文件中加载模型。以下是如何操作的示例:
保存模型:
```python
import lightgbm as lgb
# 假设你已经有了一个名为model的训练好的LightGBM模型
model.save_model('my_model.txt', format='text') # 使用文本格式保存
model.save_model('my_model.bin', format='binary') # 或者二进制格式保存(更紧凑)
```
这里的`my_model.txt`或`my_model.bin`是你指定的保存路径。
加载模型:
```python
loaded_model = lgb.Booster(model_file='my_model.txt') # 如果是文本格式
loaded_model = lgb.Booster(model_file='my_model.bin') # 如果是二进制格式
```
如果你之前知道模型的配置信息(如参数),可以提供给`Booster`构造函数:
```python
config = {'num_leaves': 31, 'learning_rate': 0.1} # 假设这是原模型的配置
loaded_model = lgb.Booster(model_file='my_model.txt', config=config)
```
这样就完成了模型的保存和加载过程。
阅读全文