sklearn模型的保存和加载API

from sklearn.externals import joblib

# 保存：joblib.dump(estimator, 'test.pkl')

# 加载：estimator = joblib.load('test.pkl')

注意：保存的后缀名是.pkl

线性回归的模型保存加载案例

保存模型

# 1.获取数据

data = load_boston()

# 2.数据集划分 数据的基本处理

x_train, x_test, y_train, y_test = train_test_split(data.data, data.target, random_state=22)

# 3.特征工程-标准化

transfer = StandardScaler()

x_train = transfer.fit_transform(x_train)

x_test = transfer.fit_transform(x_test)

# 4.1 创建模型 实例化估计器

estimator = Ridge(alpha=1)

# 4.2 训练模型 fit 正规方程计算得到最优可训练参数

estimator.fit(x_train, y_train)

#保存模型

joblib.dump(estimator,'./test.pkl')

# 加载模型

estimator = joblib.load('./test.pkl')

# 5.模型评估

y_predict = estimator.predict(x_test)

print("预测值为:\n", y_predict)

print("模型中的系数为:\n", estimator.coef_)

print("模型中的偏置为:\n", estimator.intercept_)

error = mean_squared_error(y_test, y_predict)

print("误差为:\n", error)