首页 > 数据科学 > XGBRegressor保存到本地文件并调用

XGBRegressor保存到本地文件并调用

问题:
我在python中train了一个XGBRegressor,怎么把它保存为本地文件,并且再调用呢?

答:
#训练
model = XGBRegressor()
model.fit(X, y)

#保存
from sklearn.externals import joblib
joblib.dump(model, ‘my_xgbregressor.pkl’)

#读取
model = joblib.load(‘my_xgbregressor.pkl’)

分类: 数据科学 标签: , ,
  1. 本文目前尚无任何评论.
  1. 本文目前尚无任何 trackbacks 和 pingbacks.