将已经训练好的XGBoost模型保存为.model文件
时间: 2024-06-11 17:07:04 浏览: 154
xgboost.pmml
5星 · 资源好评率100%
可以使用XGBoost提供的save_model方法将已经训练好的XGBoost模型保存为.model文件。示例如下:
```python
import xgboost as xgb
# 加载数据集
dtrain = xgb.DMatrix('train_data.csv')
dtest = xgb.DMatrix('test_data.csv')
# 定义参数
params = {
'objective': 'binary:logistic',
'eval_metric': 'auc',
'max_depth': 5,
'eta': 0.1,
'seed': 42
}
# 训练模型
model = xgb.train(params, dtrain, num_boost_round=100)
# 保存模型
model.save_model('model.model')
```
以上代码中,将已经训练好的XGBoost模型保存为.model文件,其中save_model方法的参数为文件名。保存后,可以使用xgb.Booster.load_model方法加载模型文件。
阅读全文