机器学习实践（十六）—sklearn之模型保存和加载

发表于 | 分类于 ， |
字数统计: 101

一、sklearn - 模型的保存和加载 - API

from sklearn.externals import joblib

  • 保存
    • joblib.dump(rf, ‘test.pkl’)
  • 加载
    • estimator = joblib.load(‘test.pkl’)

二、示例助解

  • 保存

    1
    2
    3
    4
    5
    6
    7
    # 使用线性模型进行预测
    # 使用正规方程求解
    lr = LinearRegression()
    # 进行训练
    lr.fit(x_train, y_train)
    # 保存训练完结束的模型
    joblib.dump(lr, "test.pkl")

  • 加载

    1
    2
    3
    # 通过已有的模型去预测
    model = joblib.load("test.pkl")
    model.predict(x_test)
本文结束 感谢您的阅读
坚持原创技术分享，您的支持将鼓励我继续创作！