机器学习代码实战——保存和加载模型（Save and Load Model）

1.实验目的

每当我们训练完一个模型后，我们需要保存训练好的模型留给下次用或者再次训练，因此我将给出两种保存模型的方法，都很简单。

数据链接
密码:35m9

2.保存与加载模型

2.1.pickle方法

import pandas as pd
import numpy as np
from sklearn.linear_model import LinearRegression  #导入线性回归模块

df = pd.read_csv('homeprices.csv')
df.head()

model = LinearRegression()
model.fit(df[['area']],df.price)    #训练模型
model.coef_    #打印斜率
model.intercept_    #打印截距
model.predict([[5000]])   #预测

2.2.joblib方法

from sklearn.externals import joblib

joblib.dump(model,'model_joblib')      #将模型写入文件model_joblib

mj = joblib.load('model_joblib')   #加载模型到mj
mj.coef_
mj.intercept_
mj.predict([[5000]])