如何使用pickle保存sklearn模型

我想使用 Pickle 转储并加载我的 Sklearn 训练模型。怎么做？

Save:

import pickle

with open("model.pkl", "wb") as f:
    pickle.dump(model, f)

Load:

with open("model.pkl", "rb") as f:
    model = pickle.load(f)

在 scikit-learn 的具体情况下，使用 joblib 可能会更好 更换pickle（转储和装载），效率更高 通常情况下，内部携带大型 numpy 数组的对象 对于拟合的 scikit-learn 估计器：

Save:

import joblib

joblib.dump(model, "model.joblib")

Load:

model = joblib.load("model.joblib")