存档

文章标签 ‘保存’

XGBRegressor保存到本地文件并调用

2018年9月2日 没有评论

问题：
我在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’)

分类: 数据科学 标签: , ,