如何使用pickle保存sklearn模型

Question

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

Answer 1

保存：

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

加载：

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

Answer 2

在所有机器学习模型中使用 pickle 都是相同的，无论类型如何，即聚类、回归等。

将模型保存在 转储中 用于 'wb' 表示写入二进制文件。

pickle.dump(model, open(filename, 'wb')) #Saving the model

在任何需要的地方加载保存的模型 load 用于 'rb' 表示读取二进制文件。

model = pickle.load(open(filename, 'rb')) #To load saved model from local directory

这里的model是kmeans，filename是本地任意文件，请按需使用。

Answer 3

也可以使用joblib

from joblib import dump, load
dump(model, model_save_path)