【sklearn】from sklearn.extermals import joblib（保存模型和加载模型）

原创博文，转载请注明出处！

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

1.载入模块

from sklearn.externals import joblib

2.保存模型

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

3.加载模型

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

4.例子

# -*- coding: utf-8 -*-
"""
# 作者：wanglei5205
# 邮箱：wanglei5205@126.com
# 博客：http://cnblogs.com/wanglei5205
# github：http://github.com/wanglei5205
"""
### 导入模块
from sklearn.datasets import load_iris
from sklearn import svm
from sklearn.externals import joblib

iris = load_iris()
x,y = iris.data,iris.target

model = svm.SVC()
model.fit(x,y)

joblib.dump(model,'dump_model.pkl')

model_new = joblib.load('dump_model.pkl')