如何使用pickle保存sklearn模型答案

【问题标题】：How to use pickle to save sklearn model如何使用pickle保存sklearn模型
【发布时间】：2019-07-19 15:18:41
【问题描述】：

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

【问题讨论】：

【解决方案1】：

保存：

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

加载：

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

【讨论】：

嗨@Mykola 你认为pickle 在保存模型方面比joblib 更好吗？因为我使用了 joblib，不幸的是，当我加载模型进行预测时，我总是得到相同的预测集群。它似乎是在保存时的模型未培训！ span>
嗨，我从未使用过joblib，无法回答您的问题。但你可以在这里询问这个问题。 span>
已经完成：stats.stackexchange.com/questions/439968/… span>

【解决方案2】：

也可以使用joblib

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

【讨论】：

【解决方案3】：

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

将模型保存在 dump 中，其中 '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

这里的模型是kmeans，文件名是任何本地文件，所以请相应地使用。

【讨论】：