如何保存和加载XGBoost模型?
时间: 2024-09-13 11:16:28 浏览: 110
XGBoost是梯度提升决策树算法的一个高效实现,广泛用于各种机器学习竞赛以及实际应用中。保存和加载XGBoost模型是模型部署和再利用的重要步骤。在Python中,XGBoost提供了简单易用的API来进行模型的保存和加载。
保存XGBoost模型通常使用`xgboost.XGBModel.save_model()`方法,该方法可以将训练好的模型保存到指定的文件路径。加载模型则使用`xgboost.XGBModel.load_model()`方法,可以从文件中读取模型。
以下是一个简单的保存和加载XGBoost模型的示例:
```python
import xgboost as xgb
# 创建并训练模型
data = xgb.DMatrix('train_data.csv?format=csv') # 加载数据
model = xgb.train(params, data, num_boost_round=10)
# 保存模型到指定路径
model.save_model('xgboost.model')
# 加载模型
loaded_model = xgb.XGBModel() # 初始化模型对象
loaded_model.load_model('xgboost.model')
# 使用加载的模型进行预测
preds = loaded_model.predict(data)
```
在这个例子中,`params`是模型训练时的参数字典,`'train_data.csv?format=csv'`是训练数据集的路径,`'xgboost.model'`是模型保存的文件名。保存模型后,您可以通过`load_model`方法加载模型文件。
阅读全文