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

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

from sklearn.externals import joblib

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

二、示例助解

  • 保存

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

    # 通过已有的模型去预测
    model = joblib.load("test.pkl")
    model.predict(x_test)
    

你可能感兴趣的:(机器学习,机器学习实践,机器学习,机器学习实践,sklearn,模型保存,模型加载)