sklearn模型训练后保存的方法

使用joblib 这个库可以将训练好的模型保存，以便下次使用。

from sklearn.datasets import load_iris
from sklearn.linear_model import LogisticRegression
from sklearn.model_selection import train_test_split
data= load_iris()
x,y=data.data,data.target
xtr,xte,ytr,yte = train_test_split(x,y,test_size=0.2)
log = LogisticRegression()
log.fit(xtr,ytr)

首先随意创建一个数据集，同时训练一个模型，查看一下模型准确率。

然后使用下面的代码保存——加载模型。就可以实现sklearn的模型存储。

#保存模型及参数
import joblib
joblib.dump(log,'my_model.pkl')
my_model_load = joblib.load('my_model.pkl')