我在python中train了一个XGBRegressor,怎么把它保存为本地文件,并且再调用呢?
1个回答
这个和保存sklearn的模型是一样的
#训练
model = XGBRegressor()
model.fit(X, y)
#保存
from sklearn.externals import joblib
joblib.dump(model, 'my_xgbregressor.pkl')
#读取
model = joblib.load('my_xgbregressor.pkl')