sklearn中保存和加载模型的方法

sklearn中保存和加载模型的方法

1.载入模块

  1 from sklearn.externals import joblib

2.保存模型

  1 joblib.dump(model,'filename.pkl')

3.加载模型

  1 model = joblib.load('filename.pkl')

4.例子

  1 # -*- coding: utf-8 -*-
  2 """
  3 # 作者：wanglei5205
  4 # 邮箱：[email protected]
  5 # 博客：http://cnblogs.com/wanglei5205
  6 # github：http://github.com/wanglei5205
  7 """
  8 ### 导入模块
  9 from sklearn.datasets import load_iris
 10 from sklearn import svm
 11 from sklearn.externals import joblib
 12 
 13 iris = load_iris()
 14 x,y = iris.data,iris.target
 15 
 16 model = svm.SVC()
 17 model.fit(x,y)
 18 
 19 joblib.dump(model,'dump_model.pkl')
 20 
 21 model_new = joblib.load('dump_model.pkl')