文章目录
一、模型的保存和载入
模型的保存和载入方便我们将训练好的模型保存在本地或发送在网上,载入模型方便我们在不同的环境下进行测试。 使用 pickle 可以进行保存与载入,也可以使用sklearn自带的函数:
import sklearn.externals as sk_externals
sk_externals.joblib.dump(model,'model.pickle') #保存
model = sk_externals.joblib.load('model.pickle') #载入
参考链接
- 十分钟上手sklearn:特征提取,常用模型,交叉验证