机器学习--模型保存和加载

保存模型和加载

语法：

from sklearn.externals import joblib

保存：

joblib.dump(rf, 'test.pkl')

rf：模型名字

test.pki：保存路径和文件名字

加载：

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

【注：文件格式pki，二进制格式】

from sklearn.linear_model import LinearRegression, SGDRegressor,  Ridge, LogisticRegression
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler
from sklearn.metrics import mean_squared_error, classification_report
from sklearn.externals import joblib
import pandas as pd
import numpy as np

def mylinear():
    """
    线性回归直接预测房子价格
    :return: None
    """
    # 获取数据
    lb = load_boston()

    # 分割数据集到训练集和测试集
    x_train, x_test, y_train, y_test = train_test_split(lb.data, lb.target, test_size=0.25)

    print(y_train, y_test)