LightGBM如何将模型保存为二进制
时间: 2024-03-23 20:43:37 浏览: 180
如何存储二进制数据方法
4星 · 用户满意度95%
要将LightGBM模型保存为二进制,可以使用LightGBM库中的`save_model()`函数。在调用该函数时,需要指定模型对象和文件路径。例如,以下代码展示了如何将模型保存为二进制:
```python
import lightgbm as lgb
# 训练模型
train_data = lgb.Dataset(X_train, label=y_train)
params = {'objective': 'binary', 'metric': 'auc'}
model = lgb.train(params, train_data)
# 将模型保存为二进制
model.save_model('model.bin')
```
在上面的代码中,`model.save_model('model.bin')`将模型保存为二进制文件`model.bin`。保存后,可以使用`lgb.Booster()`函数从文件中加载模型,例如:
```python
# 加载模型
loaded_model = lgb.Booster(model_file='model.bin')
# 使用模型进行预测
y_pred = loaded_model.predict(X_test)
```
阅读全文