一、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)