机器学习------模型的保存与加载

API:

from sklearn.externals import joblib

机器学习------模型的保存与加载_第1张图片

保存:

    # estimator测试
    # ①正规方程求解
    lr = LinearRegression()
    lr.fit(x_train, y_train)
    # 模型保存
    joblib.dump(lr, "./text.pkl")

加载:

    # 模型加载
    llr = joblib.load("./text.pkl")
    # 加载的模型使用
    y_llr_predict = sd_y.inverse_transform(llr.predict(x_test))
    print("真实价格:", sd_y.inverse_transform(y_test))
    print("4444444444444444444444444444444444444")
    print("正规方程测试集预测结果:", y_llr_predict)
    print("4444444444444444444444444444444444444")

 

 

 

 

 

 

 

你可能感兴趣的:(机器学习)