Xgboost 模型保存和载入()

https://blog.csdn.net/u012884015/article/details/78653178

xgb_model.get_booster().save_model('xgb.model')
tar = xgb.Booster(model_file='xgb.model')
x_test = xgb.DMatrix(x_test)
pre=tar.predict(x_test)
act=y_test
print(mean_squared_error(act, pre))

  

猜你喜欢

转载自www.cnblogs.com/Allen-rg/p/9477793.html